List Question
10 TechQA 2025-01-02 04:06:10Optax equivalent of flax's optim.Adam.apply_gradient()
149 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?
634 views
Asked by Luca Thiede
Pickle changes type in jax
460 views
Asked by Valentin Macé
Calculating the Hessian Vector Product of a Flax NN output wrt to the inputs
440 views
Asked by Vignesh Gopakumar
Getting incorrect output from the flax model's init call
293 views
Asked by Bunny Rabbit
How to convert .safetensors or .ckpt Files and Using in FlaxStableDiffusionImg2ImgPipeline?
3.1k views
Asked by Aero Wang
Computing dot product of gradients with itself for a neural network model in JAX
199 views
Asked by abc
Computing the gradient of a batched function using JAX
617 views
Asked by al_cc
Prefetching an iterator of 128-dim array to device
141 views
Asked by jeffreyveon