Question

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

score 139 · Answer 1 · 2025-01-02 04:06:10

139

Views

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

Published on 02 January 2025 at 04:06

score 7.8k · Answer 2 · 2025-01-02 04:09:17

7.8k

Views

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

Published on 02 January 2025 at 04:09

score 623 · Answer 3 · 2025-01-02 04:06:09

623

Views

Flax much slower than pure Jax for neural nentworks?

Published on 02 January 2025 at 04:06

score 448 · Answer 4 · 2025-01-02 04:02:08

448

Views

Pickle changes type in jax

Published on 02 January 2025 at 04:02

score 431 · Answer 5 · 2025-01-02 04:09:50

431

Views

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

Published on 02 January 2025 at 04:09

score 283 · Answer 6 · 2025-01-03 08:46:34

283

Views

Getting incorrect output from the flax model's init call

Published on 03 January 2025 at 08:46

score 3k · Answer 7 · 2025-01-02 04:10:58

3k

Views

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

Published on 02 January 2025 at 04:10

score 191 · Answer 8 · 2025-01-02 04:10:17

191

Views

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

Published on 02 January 2025 at 04:10

score 609 · Answer 9 · 2025-01-02 04:06:49

609

Views

Computing the gradient of a batched function using JAX

Published on 02 January 2025 at 04:06

score 129 · Answer 10 · 2025-01-02 04:06:08

129

Views

Prefetching an iterator of 128-dim array to device

Published on 02 January 2025 at 04:06

score 140 · Answer 11 · 2025-01-02 04:10:16

140

Views

How to load checkpoints to flax state in jax 0.4.14 when the save path is changed?

Published on 02 January 2025 at 04:10

score 331 · Answer 12 · 2025-01-02 04:05:29

331

Views

Pytorch equivalent of `register_buffer` in flax/jax

Published on 02 January 2025 at 04:05

score 650 · Answer 13 · 2025-01-02 04:04:44

650

Views

How to build a Pytorch-like code in Jax Flax

Published on 02 January 2025 at 04:04

score 82 · Answer 14 · 2025-01-02 04:10:15

82

Views

Fail to understand the usage of partial argument in Flax Resnet Official Example

Published on 02 January 2025 at 04:10

score 1.9k · Answer 15 · 2025-01-02 04:11:19

1.9k

Views

No module named 'jax.experimental.global_device_array' when running the official Flax Example on Colab with V100

Published on 02 January 2025 at 04:11

score 270 · Answer 16 · 2025-01-02 04:10:14

270

Views

How to unroll the training loop so that Jax can train multiple steps in GPU/TPU

Published on 02 January 2025 at 04:10

score 42 · Answer 17 · 2025-01-02 04:07:30

42

Views

Data download issue with official Flax Image Net example

Published on 02 January 2025 at 04:07

score 52 · Answer 18 · 2025-01-02 04:02:52

52

Views

The exact meaning of n_jitted_steps=5

Published on 02 January 2025 at 04:02

score 61 · Answer 19 · 2025-01-02 04:09:17

61

Views

Passing JAX tracers to Huggingface CLIP transformer for calculating loss

Published on 02 January 2025 at 04:09

score 46 · Answer 20 · 2025-01-02 04:02:09

46

Views

data_format in JAX/FLAX

Published on 02 January 2025 at 04:02

List Question

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

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

Flax much slower than pure Jax for neural nentworks?

Pickle changes type in jax

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

Getting incorrect output from the flax model's init call

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

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

Computing the gradient of a batched function using JAX

Prefetching an iterator of 128-dim array to device

How to load checkpoints to flax state in jax 0.4.14 when the save path is changed?

Pytorch equivalent of `register_buffer` in flax/jax

How to build a Pytorch-like code in Jax Flax

Fail to understand the usage of partial argument in Flax Resnet Official Example

No module named 'jax.experimental.global_device_array' when running the official Flax Example on Colab with V100

How to unroll the training loop so that Jax can train multiple steps in GPU/TPU

Data download issue with official Flax Image Net example

The exact meaning of n_jitted_steps=5

Passing JAX tracers to Huggingface CLIP transformer for calculating loss

data_format in JAX/FLAX

Trending Questions

Popular # Hahtags

Popular Questions