jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
How to generate random numbers between 0 and 1 in jax?
user_2361060
•
3
1
455
jax
Taking rows of a matrix given a batch of indices - Python
user_142026060
•
2
2
52
jax rows matrix python
JAX return types after transformations
user_186499920
•
1
1
100
autograd jax jit python
Accumulation in JAX
user_186499920
•
1
1
341
xla accumulate jax jit python
Optimizing sampling with varying sample sizes in jax
user_189391710
•
1
0
28
google-jax jax random python
jax linear solve issues
user_170908510
•
3
2
433
jax linear-algebra python
JAX pytree for xarray
user_186499920
•
1
1
278
python-xarray jax numpy python
inserting rows and columns of zeros to a sparse array in python
user_188685740
•
1
2
143
jax scipy sparse-matrix numpy python
jax empty element in the parameters of neural network
user_136049120
•
1
1
46
autograd jax
Nested for loops in jax
user_40144870
•
1
1
187
jax jit python
Hot Questions