jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
Multiple `vmap` in JAX?
user_150051030
•
1
2
700
jax vectorization python
JIT a least squares loss function in Jax
user_8961120
•
2
1
397
jax python
Inverse operation to padding in Jax
user_170814890
•
2
1
121
jax pytorch padding
How to use jax vmap for nested loops?
user_134681280
•
5
1
517
jax vectorization jit python performance
JAX/JIT vs Std Numpy performance: where I am wrong?
user_116356540
•
2
1
453
jax jit numpy python
Jax vectorization: vmap and/or numpy.vectorize?
user_116356540
•
4
1
1214
jax python
How to reduce JAX compile time when using for loop?
user_134681280
•
3
2
1857
jax jit python loops performance
Why does Mypy think adding two Jax arrays returns a numpy array?
user_16866290
•
2
2
412
python-typing jax mypy type-hinting python
Websockets messages only sent at the end and not in instances using async / await, yield in nested for loops
user_36611860
•
3
2
729
jax fastapi async-await websocket python
Is there a way to disable forward evaluation while using VJP in JAX?
user_100560660
•
2
1
62
jax python
Hot Questions