Göktuğ Güvercin's repositories
000
dm-haiku
JAX-based neural network library
Language:PythonApache-2.0000
Language:Jupyter Notebook000
Language:Python000
flax
Flax is a neural network library for JAX that is designed for flexibility.
Apache-2.0000
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Apache-2.0000
JAX-Tutorials
The fundamentals of Google's last ML framework JAX are explained.
MIT000
Language:Python000