DEVHIDE
Login Or Sign up

List Question

20
121
Views

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

Published on
#python #deep-learning #jax #mlp #flax
7.8k
Views

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

Published on
#deep-learning #neural-network #jax #flax #dm-haiku
600
Views

Flax much slower than pure Jax for neural nentworks?

Published on
#python #jax #flax
428
Views

Pickle changes type in jax

Published on
#python #numpy #pickle #jax #flax
424
Views

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

Published on
#jax #hessian #automatic-differentiation #flax
262
Views

Getting incorrect output from the flax model's init call

Published on
#jax #flax
3k
Views

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

Published on
#python #pytorch #stable-diffusion #flax
184
Views

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

Published on
#deep-learning #neural-network #jax #flax
593
Views

Computing the gradient of a batched function using JAX

Published on
#machine-learning #jax #flax
110
Views

Prefetching an iterator of 128-dim array to device

Published on
#python #numpy #jax #flax
150
Views

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

Published on
#jax #flax
335
Views

Pytorch equivalent of `register_buffer` in flax/jax

Published on
#python #deep-learning #pytorch #jax #flax
648
Views

How to build a Pytorch-like code in Jax Flax

Published on
#python #machine-learning #pytorch #jax #flax
81
Views

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

Published on
#jax #flax
1.9k
Views

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

Published on
#jax #flax
272
Views

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

Published on
#jax #flax
41
Views

Data download issue with official Flax Image Net example

Published on
#tensorflow-datasets #jax #flax
51
Views

The exact meaning of n_jitted_steps=5

Published on
#jax #flax
59
Views

Passing JAX tracers to Huggingface CLIP transformer for calculating loss

Published on
#huggingface-transformers #clip #jax #flax
44
Views

data_format in JAX/FLAX

Published on
#jax #flax

Trending Questions

Popular # Hahtags

javascript python java c# php android html jquery c++ css ios sql mysql r reactjs node.js arrays c asp.net json python-3.x ruby-on-rails .net sql-server swift django angular objective-c pandas excel

Popular Questions

Copyright © 2021 Jogjafile Inc.