flax

7 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
AttributeError: module 'flax' has no attribute 'optim'
user_196856410
•
2
1
316
flax google-colaboratory python-3.x
How to build a Pytorch-like code in Jax Flax
user_196341570
•
1
0
95
flax jax pytorch machine-learning python
Pytorch equivalent of `register_buffer` in flax/jax
user_22270260
•
1
0
47
flax jax pytorch deep-learning python
is there a way to trace grads through self.put_variable method in flax?
user_121393490
•
1
2
53
flax jax python
Calculating the Hessian Vector Product of a Flax NN output wrt to the inputs
user_63694510
•
2
1
90
hessian flax jax automatic-differentiation
Flax much slower than pure Jax for neural nentworks?
user_38694480
•
3
1
254
flax jax python
What is the main difference between flax (google) and dm-haiku (deepmind)?
user_39600380
•
10
1
3936
dm-haiku flax jax deep-learning neural-network
  • 1
Hot Questions