List Question
10 TechQA 2025-01-02 04:06:10Optax equivalent of flax's optim.Adam.apply_gradient()
112 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?
600 views
Asked by Luca Thiede
Pickle changes type in jax
422 views
Asked by Valentin Macé
Calculating the Hessian Vector Product of a Flax NN output wrt to the inputs
421 views
Asked by Vignesh Gopakumar
Getting incorrect output from the flax model's init call
259 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
178 views
Asked by abc
Computing the gradient of a batched function using JAX
583 views
Asked by al_cc
Prefetching an iterator of 128-dim array to device
106 views
Asked by jeffreyveon