TechQA.

Question

Optax equivalent of flax's optim.Adam.apply_gradient()

score 117 · Answer 1 · 2024-09-16 20:09:19

Optax equivalent of flax's optim.Adam.apply_gradient()

score 7844 · Answer 2 · 2024-09-18 13:36:08

What is the main difference between flax (google) and dm-haiku (deepmind)?

score 596 · Answer 3 · 2024-09-16 06:32:07

Flax much slower than pure Jax for neural nentworks?

score 423 · Answer 4 · 2024-09-17 14:48:27

423 views Asked by Valentin Macé At 17 September 2024 at 14:48

score 420 · Answer 5 · 2024-09-16 15:11:55

Calculating the Hessian Vector Product of a Flax NN output wrt to the inputs

score 257 · Answer 6 · 2024-09-16 09:54:53

Getting incorrect output from the flax model's init call

score 3068 · Answer 7 · 2024-09-19 07:15:51

How to convert .safetensors or .ckpt Files and Using in FlaxStableDiffusionImg2ImgPipeline?

score 179 · Answer 8 · 2024-09-16 23:29:53

Computing dot product of gradients with itself for a neural network model in JAX

score 589 · Answer 9 · 2024-09-19 09:39:24

Computing the gradient of a batched function using JAX

score 105 · Answer 10 · 2024-09-15 19:17:29

Prefetching an iterator of 128-dim array to device

List Question