Google's functional ML library with XLA compilation. The framework that trained Gemini, PaLM, and many other Google models.
Open source ↗