jax

jax

9 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
richinex0
•
1
1
39
minimization jax mathematical-optimization math python
what can casue "LLVM ERROR: Trying to register different dialects for the same namespace"
ipe_zyz0
•
0
0
20
jax xla llvm tensorflow
Jax - vmap over batch of dataclasses
EmptyJackson0
•
0
0
28
flax jax python
Where is the batch dimension with jax.pmap?
Noel Tong0
•
0
0
18
jax python
module 'jax' has no attribute 'tree_multimap' in AlphaFold2 CoLab
Dr. Wilson0
•
2
1
431
haiku jax google-colaboratory python
How to write this function jax.jit-able>
Nhật Minh Lê0
•
1
1
36
jax
If there are two functions - one with jit and other without, and when I iterate them for 100 times, unjit function gives me a less time than jit one
Yagnesh Bhadiyadra0
•
-1
1
45
jit jax google-colaboratory time python
How to map the kronecker product along array dimensions?
bayes20210
•
2
1
51
kronecker-product jax numpy arrays python
JAX - jitting functions: parameters vs "global" variables
Liuka0
•
1
1
66
jit jax deep-learning python
  • 1
Hot Questions