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
Flux multi-cpu parallelism? New to Julia question , flux , zygote	9	2967	November 21, 2020
[blog post] Implement your own AD with Julia in ONE day Community blog-post	33	4377	November 3, 2018
Kron vs scalar product speed difference. python code faster? New to Julia question	41	4243	January 14, 2017
Comparison of automatic differentiation tools from 2016 still accurate? Numerics differentiation	41	5940	August 16, 2018
Thoughts on JAX vs Julia Community jax	21	11480	October 25, 2022