jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
What is the fastest way of selecting a subset of a JAX matrix?
user_8660820
•
1
1
416
jax python
How do I save an optimizer state of JAX trained model?
user_141322150
•
2
1
1518
jax machine-learning python
Why is this function slower in JAX vs numpy?
user_131913050
•
5
2
3641
jax numpy optimization python performance
Conditional update in JAX?
user_748650
•
3
1
1958
autograd jax python
Efficient way to compute Jacobian x Jacobian.T
user_2982090
•
2
1
908
backpropagation autograd jax pytorch python
Not able to install jaxlib
Admin1
•
8
5
20702
jax pip installation numpy python
Get and Post API call in java with basic authentication
user_98936770
•
3
1
7027
jax jax-rs java
Google JAX 1D convolutional neural network
user_131122230
•
3
1
482
jax conv-neural-network python
What is the difference between JAX, Trax, and TensorRT, in simple terms?
user_61068420
•
5
1
1337
trax jax tensorflow
Can pytorch optimize sequential operations (like a tensorflow graph or JAX's jit)?
user_4976000
•
2
1
843
jax pytorch python
Hot Questions