List Question
10 TechQA 2025-01-02 04:06:10Optax equivalent of flax's optim.Adam.apply_gradient()
131 views
Asked by CV_enth
What is the main difference between flax (google) and dm-haiku (deepmind)?
7.8k views
Asked by discort
Flax much slower than pure Jax for neural nentworks?
615 views
Asked by Luca Thiede
Pickle changes type in jax
441 views
Asked by Valentin Macé
Calculating the Hessian Vector Product of a Flax NN output wrt to the inputs
425 views
Asked by Vignesh Gopakumar
Getting incorrect output from the flax model's init call
276 views
Asked by Bunny Rabbit
How to convert .safetensors or .ckpt Files and Using in FlaxStableDiffusionImg2ImgPipeline?
3k views
Asked by Aero Wang
Computing dot product of gradients with itself for a neural network model in JAX
183 views
Asked by abc
Computing the gradient of a batched function using JAX
599 views
Asked by al_cc
Prefetching an iterator of 128-dim array to device
122 views
Asked by jeffreyveon