tpu

A high-throughput and memory-efficient inference and serving engine for LLMs

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

Fast and flexible AutoML with learning guarantees.

Large-scale LLM inference engine

DECIMER Image Transformer is a deep-learning-based tool designed for automated recognition of chemical structure images. Leveraging transformer architectures, the model converts chemical images into SMILES strings, enabling the digitization of chemical data from scanned documents, literature, and patents.

Official implementation of 'Transformer-VQ: Linear-Time Transformers via Vector Quantization' (ICLR 2024)

Tutorial to pretrain & fine-tune a 🤗 Flax T5 model on a TPUv3-8 with GCP

:dart: Gradient Accumulation for TensorFlow 2

Train GEMMA on TPU/GPU! (Codebase for training Gemma-Ko Series)

🤗 HuggingFace Diffusers Flax TPU and PyTorch GPU for Colab