Software Engineer at Google DeepMind working on JAX/Flax
No description provided.
A functional training loops library for JAX
Concurrent data pipelines in Python >>>
A simple, performant and scalable Jax LLM!