#optax - Wesley Kambale

Skip to main content

Hashnode Wesley Kambale

Machine Learning
JAX Series
Speaker Profile

Command Palette

Search for a command to run...

#optax

Articles tagged with #optax

Custom training loop from scratch in JAX
Writing a training loop that you can actually control
Mar 4, 202610 min read166

© 2026 Wesley Kambale

Archive
Privacy
Terms