jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
Flax much slower than pure Jax for neural nentworks?
user_38694480
•
3
1
254
flax jax python
Cannot install Jaxlib on Apple M1, on Docker
user_53862990
•
2
0
518
apple-m1 jax docker gcc python
How could I speed up this looping code by JAX; Finding nearest neighbors for collision
user_133948170
•
2
2
1078
jax jit python
swapaxes and how it is implemented?
user_107134280
•
1
1
67
jax arrays python
Error upon compilation while using jax.jit
user_131255810
•
1
1
241
jax python
Resulting array is zero when jax.numpy is used
user_131255810
•
1
2
161
jax numpy python
TypeError: Argument 'MLP( # attributes num_neurons_per_layer = [4, 1] )' of type <class '__main__.MLP'> is not a valid JAX type
user_110654150
•
1
1
254
jax python
Jacobian diagonal computation in JAX
user_56289350
•
1
1
220
jax automatic-differentiation deep-learning python
How to install trax, jax, jaxlib on M1 Mac on macOS 12?
user_5838340
•
3
2
3774
trax apple-m1 jax anaconda python
JAX: Converting Concrete Tracer values to regular float values doesnt work
user_153423550
•
3
0
516
jax python
Hot Questions