Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
updated at May 26, 2024, 8:28 p.m.
9 +0
158 -1
18 +1
A Graph Neural Network Library in Jax
updated at May 26, 2024, 8:34 p.m.
41 +0
1,329 +3
87 +0
A differentiable cosmology library in JAX
updated at May 26, 2024, 8:35 p.m.
18 +0
159 +2
31 +1