jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
Differences in the standard error of parameters computed via the hessian inverse and via QR decomposition
user_131255810
•
1
1
39
minimization jax mathematical-optimization math python
module 'jax' has no attribute 'tree_multimap' in AlphaFold2 CoLab
user_199793490
•
2
1
431
haiku jax google-colaboratory python
How to write this function jax.jit-able>
user_199790860
•
1
1
36
jax
How to map the kronecker product along array dimensions?
user_195429080
•
2
1
51
kronecker-product jax numpy arrays python
JAX - jitting functions: parameters vs "global" variables
user_31601860
•
1
1
66
jax deep-learning jit python
Jax.lax.scan with arguments?
user_196953820
•
1
1
64
google-jax jax
Rewriting for loop with jax.lax.scan
user_196953820
•
1
1
69
google-jax jax python
Conceptual difference between TensorFlow (graph-mode) and JAX
user_1333740
•
2
0
63
jax tensorflow
JAX: Why not using @jit yields -inf value but using it doesn't?
user_124969370
•
3
1
48
jax python
JAX grad function: why am I getting a list of zeros instead of the gradients?
user_196953820
•
1
1
70
google-jax jax automatic-differentiation mathematical-optimization python
Hot Questions