Julia Programming Language
`Zygote.gradient` is 54000 TIMES slower than `jax.gradient`
Specific Domains
Optimization (Mathematical)
zygote
,
jax
yolhan_mannes
January 30, 2025, 8:57pm
43
Yes I did that twice I think, I will avoid it promise
show post in topic
Related topics
Topic
Replies
Views
Activity
Zygote much slower than JAX for automatic differentiation of energy
Machine Learning
performance
,
flux
,
zygote
,
jax
,
lux
22
1749
May 15, 2024
Any faster way of computing small gradients?
Performance
zygote
,
forwarddiff
,
symbolics
,
autodiff
21
2143
August 11, 2022
Speeding up Zygote autodiff for numerical loop
Performance
question
13
404
December 16, 2024
Lux, ComponentArrays and flat parameters : computing the gradient works with Zygote but not with Enzyme
New to Julia
enzyme
16
1895
May 14, 2024
Performance comparison - Flux.jl's Adam vs Jax's Adam
Performance
question
,
package
,
performance
,
flux
33
4147
November 3, 2022