jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
JAX Apply function only on slice of array under jit
user_87370160
•
4
2
1346
jax numpy python
Debug array in jax vmap function
user_116356540
•
3
1
714
jax vectorization python
sum matrix elementwise using vmap (jax)?
user_57913980
•
2
1
550
jax python
importing jax fails on mac with m1 chip
user_132807480
•
5
4
3117
apple-m1 jax python-3.8 python
Not able to import python package jax in Google TPU
user_3958820
•
3
2
787
spacy-transformers tpu sentence-transformers jax python-3.x
JAX batching with different lengths
user_87370160
•
2
1
730
jax numpy python
Jax - autograd of a sigmoid always returns nan
user_17606150
•
2
1
673
jax nan python
How to install just XLA?
user_59869070
•
2
2
214
tensorflow-xla jax tensorflow2.0 tensorflow
JAX does jitting functions separately change performances?
user_87370160
•
2
1
136
jax jit
Gradient Accumulation with JAX
user_60842450
•
2
1
543
gradient-descent jax python
Hot Questions