Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
created at April 9, 2023, 5:18 p.m.
9 +0
158 -1
18 +1
LoRA for arbitrary JAX models and functions
created at April 21, 2023, 11:50 a.m.
3 +0
119 +0
4 +0
XLB: Accelerated Lattice Boltzmann (XLB) based on JAX for Physics-based ML
created at May 4, 2023, 5:36 p.m.
10 +0
147 +1
14 -1
Spyx: Spiking Neural Networks in JAX
created at June 21, 2023, 8:42 p.m.
78 +1