jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
what exactly is `xla_client` in the jax library?
user_131592870
•
1
1
404
jax tensorflow
Create a 3D tensor of zeros with exactly one '1' randomly placed on every slice in numpy/jax
user_19700530
•
5
3
338
jax tensor numpy python
Non-hashable static arguments are not supported in Jax when using vmap
user_147868130
•
7
1
2606
jax python
Jax cannot find the static argnums
user_147868130
•
3
1
1059
jax python
Unable to Install Specific JAX jaxlib GPU version
user_45704720
•
2
3
4660
jax failed-installation gpu installation
Saving Gradient in Backward Pass Google-JAX
user_20787960
•
4
0
304
google-jax backpropagation jax gradient machine-learning
Jaxlib pip installation failure
user_131422450
•
2
1
1859
jax pip windows python
Can OpenMDAO co-operate with autograd or jax?
user_146219530
•
2
0
108
autograd openmdao jax
Performance drop when slicing jax.numpy arrays
user_32537810
•
3
2
642
jax scipy jit python performance
Nontransitive subclassing with numpy and jax
user_145671350
•
3
1
144
jax numpy python
Hot Questions