jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
Parameters do not converge at a lower tolerance in nonlinear least square implementation in python
user_10828830
•
1
1
62
convergence non-linear-regression jax python
vmap in Jax to loop over arguments
user_181423290
•
2
1
307
jax arrays python
DenseElementsAttr could not be constructed from the given buffer
user_187700150
•
2
1
364
jax
JAX(XLA) vs Numba(LLVM) Reduction
user_186499920
•
2
1
1541
numba jax jit reduce python
Handle varying shapes in jax numpy arrays (jit compatible)
user_98176770
•
2
1
476
jax jit shapes arrays python
jax: sample many observations from random.choice with replacement between them
user_4976000
•
1
1
110
jax
Output shape of jax's vmap
user_117189470
•
1
2
181
jax python
How to print with JAX
user_133948170
•
1
1
1067
jax jit
How to iterate over tuples using jax.lax.scan
user_44345490
•
1
1
562
jax python
Mask a numpy array after a given value
user_98176770
•
2
3
223
jax mask numpy sum python
Hot Questions