Exploring hyperparameter meta-loss landscapes with JAX by Luke Metz
Demonstrates how to use JAX to perform inner-loss optimization with SGD and Momentum, outer-loss optimization with gradients, and outer-loss optimization using evolutionary strategies.
- from
- JAX
- added
- 2026-10-10
- likes
- 0
similar
-
Meta-Learning in 50 Lines of JAX by Eric Jang blog.evjang.com
Introduction to both JAX and Meta-Learning.
-
Evolving Neural Networks in JAX by Robert Tjarko Lange roberttlange.github.io
Explores how JAX can power the next generation of scalable neuroevolution algorithms.
-
Jax in Action manning.com
A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.
-
Using JAX to accelerate our research by David Budden and Matteo Hessel deepmind.com
Describes the state of JAX and the JAX ecosystem at DeepMind.
-
-
JAX: Accelerated Machine Learning Research | SciPy 2020 | VanderPlas youtube.com
JAX's core design, how it's powering new research, and how you can start using it.
JAX › Tutorials and Blog Posts: “Demonstrates how to use JAX to perform inner-loss optimization with SGD and Momentum, outer-loss optimization with gradients, and outer-loss optimization using evolutionary strategies.”