Popular GitHub Repositories for Tpus

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, JAXADi enables the creation of efficient code that runs smoothly on CPUs, GPUs, and TPUs.

? Training StyleGAN2 on TPUs in JAX

? The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX

Train DCGAN with TPUs on Google Cloud

A package for fine-tuning Transformers with TPUs, written in Tensorflow2.0+

Automatically take good care of your preemptible TPUs

BiSeNet V2 TensorFlow 2 Implementation capable of running on Edge TPUs

Reasoning-from-Zero using gemma.JAX.nnx on TPUs

Tutorial for text classification with fine tuning of a T5 model on TPUs.

⚡Powered by Goggle TPUs and the latest (Aug 27, 2024) Gemini 1.5 Pro and Flash Models to generate high-quality engineered prompts, analyze text and images, and create datasets for fine-tuning AI models, helping you to become a prompt engineering pro

Python package to help train image classification deep neural networks for generic datasets. Especially powerful when using with TPUs. The main advantages of the package is that is leverages the tfrecords format for the data along with transfer learning and hyperparameter optimization for model training.

Classification of multilingual dataset trained only on English training data using pre-trained models. Model is trained on TPUs using PyTorch and torch_xla library.

Pre-training GPT-2 model from scratch using GPUs and TPUs.

In this project, we will use Google Colab for model training and run the Tensorflow1.15 own object detection model. Colab is a free Jupyter Notebook environment hosted by Google that runs on the cloud. Google Colab provides free access to GPUs (Graphical Processing Units) and TPUs (Tensor Processing Units).