jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
Recursion in JAX
user_11529770
•
1
1
48
jax recursion python
Local devices VS non local devices in multi GPU processing
user_73396240
•
1
1
35
multi-gpu jax deep-learning gpu multiprocessing
How to specify or set a variable to a GPU device
user_73396240
•
1
1
78
multi-gpu jax deep-learning gpu python
JAX: average per-example gradients different from aggregated gradients
user_38617750
•
2
1
58
jax python
JAX GPU memory usage even with CPU allocation
user_197440780
•
2
0
27
jax gpu python
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
JAX Tridiagonal Jacobians
user_186499920
•
1
0
103
tensorflow-xla jax automatic-differentiation jit python
Is it safe to read the value of numpy.empty or jax.numpy.empty?
user_53997340
•
4
1
74
jax language-lawyer memory numpy c
What is the recommended way to do embeddings in jax?
user_2872380
•
2
1
102
jax
Hot Questions