Question

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

score 102 · Answer 1 · 2023-11-13 06:14:20

102

Views

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

Published on 13 November 2023 at 06:14

score 7.8k · Answer 2 · 2021-04-29 14:25:51

7.8k

Views

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

Published on 29 April 2021 at 14:25

score 589 · Answer 3 · 2022-03-05 00:40:05

589

Views

Flax much slower than pure Jax for neural nentworks?

Published on 05 March 2022 at 00:40

score 412 · Answer 4 · 2022-05-10 17:58:25

412

Views

Pickle changes type in jax

Published on 10 May 2022 at 17:58

score 414 · Answer 5 · 2022-06-02 16:17:35

414

Views

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

Published on 02 June 2022 at 16:17

score 249 · Answer 6 · 2022-12-04 12:20:21

249

Views

Getting incorrect output from the flax model's init call

Published on 04 December 2022 at 12:20

score 3k · Answer 7 · 2023-06-24 06:41:57

3k

Views

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

Published on 24 June 2023 at 06:41

score 172 · Answer 8 · 2023-06-27 02:07:06

172

Views

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

Published on 27 June 2023 at 02:07

score 573 · Answer 9 · 2023-07-28 21:54:45

573

Views

Computing the gradient of a batched function using JAX

Published on 28 July 2023 at 21:54

score 98 · Answer 10 · 2023-08-05 16:28:10

98

Views

Prefetching an iterator of 128-dim array to device

Published on 05 August 2023 at 16:28

score 132 · Answer 11 · 2023-08-10 03:32:57

132

Views

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

Published on 10 August 2023 at 03:32

score 325 · Answer 12 · 2022-08-07 09:58:11

325

Views

Pytorch equivalent of `register_buffer` in flax/jax

Published on 07 August 2022 at 09:58

score 642 · Answer 13 · 2022-08-09 19:57:05

642

Views

How to build a Pytorch-like code in Jax Flax

Published on 09 August 2022 at 19:57

score 75 · Answer 14 · 2023-05-05 00:24:20

75

Views

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

Published on 05 May 2023 at 00:24

score 1.9k · Answer 15 · 2023-05-07 01:57:58

1.9k

Views

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

Published on 07 May 2023 at 01:57

score 263 · Answer 16 · 2023-05-07 02:47:46

263

Views

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

Published on 07 May 2023 at 02:47

score 34 · Answer 17 · 2023-05-08 00:24:49

34

Views

Data download issue with official Flax Image Net example

Published on 08 May 2023 at 00:24

score 44 · Answer 18 · 2023-05-08 02:06:02

44

Views

The exact meaning of n_jitted_steps=5

Published on 08 May 2023 at 02:06

score 53 · Answer 19 · 2023-05-15 21:59:33

53

Views

Passing JAX tracers to Huggingface CLIP transformer for calculating loss

Published on 15 May 2023 at 21:59

score 35 · Answer 20 · 2023-06-06 10:00:23

35

Views

data_format in JAX/FLAX

Published on 06 June 2023 at 10:00

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