jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
Compute efficiently Hessian matrices in JAX
user_38617750
•
3
1
780
jax
in_axes keyword in JAX's vmap
user_38617750
•
4
1
505
auto-vectorization jax
Differentiation of an improper integral using JAX and SciPy
user_10972000
•
2
2
311
jax automatic-differentiation scipy python
Fastest way to multiply and sum 4D array with 2D array in python?
user_123429350
•
4
1
98
jax linear-algebra numpy arrays python
Is there a module to convert a tensorflow NN to Jax?
user_147203800
•
2
2
354
jax tensorflow python
How to deal with ImportError: cannot import name 'pytree' from 'jaxlib' of using jax?
user_176433400
•
1
1
1053
jax importerror ubuntu python
What is the alternative of elementwise_grad of autograd in JAX?
user_176263980
•
1
1
281
jax automatic-differentiation neural-network python
jax confusion - what are the best practices?
user_29772560
•
1
1
570
jax numpy python
Automatic Differentiation with respect to rank-based computations
user_141128250
•
3
1
91
autodiff autograd jax automatic-differentiation tensorflow
Python - time difference (JAX library)
user_162885200
•
2
1
203
jax python-3.x
Hot Questions