Language model alignment-focused deep learning curriculum

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…

Flax is a neural network library for JAX that is designed for flexibility.

Implementation of Dynamic Routing Between Capsules

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Enabling PyTorch on XLA Devices (e.g. Google TPU)