Performance comparison - Flux.jl's Adam vs Jax's Adam

richinex October 31, 2022, 7:56pm 20

Yea I’m still cracking my brains on how best to work around it.

Topic		Replies	Views
`Zygote.gradient` is 54000 TIMES slower than `jax.gradient` Optimization (Mathematical) zygote , jax	80	1859	February 1, 2025
Zygote much slower than JAX for automatic differentiation of energy Machine Learning performance , flux , zygote , jax , lux	22	1712	May 15, 2024
Thoughts on JAX vs Julia Community jax	21	11796	October 25, 2022
Flux multi-cpu parallelism? New to Julia question , flux , zygote	9	3017	November 21, 2020
State of machine learning in Julia Machine Learning	60	66586	August 26, 2022