Google DeepMind. Gemini.
-
Google
- Mountain View, CA
- https://sites.google.com/corp/view/yxu/
Stars
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…