jax

130 Questions
Newest Active Unanswered Frequent
Score
View
Card Compact
Efficient way of "graph based" polynomial evaluation with different coefficients
user_43605570
•
1
1
57
piecewise jax tensorflow vectorization branch
Best way to compute the moving average of word vectors in JAX
user_60842450
•
2
1
87
jax python
is it possible to jit a function which uses jax.numpy.unique?
user_4976000
•
3
1
453
jax jit
Jax vmap for simple array update
user_160043560
•
2
1
248
jax python
Why my neural network does not converge using Jax？
user_112042470
•
2
1
216
jax neural-network python
What is the main difference between flax (google) and dm-haiku (deepmind)?
user_39600380
•
10
1
3936
dm-haiku flax jax deep-learning neural-network
What is JaxNumpy-compatible equivalent to this Python function?
user_2904430
•
2
1
207
jax numpy python
why when i have a np.power in my function jax.grad can't give me the derivitives?
user_66637330
•
4
1
630
jax linear-regression numpy python
Jax, jit and dynamic shapes: a regression from Tensorflow?
user_99738790
•
4
2
903
jax tensorflow python
JAX vmap behaviour
user_39524240
•
2
1
475
jax vectorization python
Hot Questions