Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
☆64Jul 22, 2026Updated last month
Alternatives and similar repositories for ml-testing-accelerators
Users that are interested in ml-testing-accelerators are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆16Mar 13, 2025Updated last year
- This repository contains example code to build models on TPUs☆30Feb 17, 2023Updated 3 years ago
- ☆200Updated this week
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆85Dec 18, 2025Updated 8 months ago
- Inference code for LLaMA models☆20Apr 3, 2025Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- A utility for stressing GPUs by driving utilization (and thus power consumption) up and down in user-defined cycle intervals. It will als…☆26May 10, 2023Updated 3 years ago
- ☆32Updated this week
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,805May 27, 2026Updated 3 months ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆558Aug 7, 2026Updated 3 weeks ago
- PyTorch centric eager mode debugger☆46Dec 16, 2024Updated last year
- ☆34Jun 3, 2024Updated 2 years ago
- 🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.☆17Jun 5, 2025Updated last year
- torchprime is a reference model implementation for PyTorch on TPU.☆49Mar 3, 2026Updated 5 months ago
- "Not too complicated" training code for CIFAR-10 by PyTorch Lightning☆12Jun 5, 2022Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Fast and easy distributed model training examples.☆12Nov 26, 2024Updated last year
- TPU inference for vLLM, with unified JAX and PyTorch support.☆420Updated this week
- Tokamax: A GPU and TPU kernel library.☆274Updated this week
- ☆22Dec 14, 2021Updated 4 years ago
- This repository contains code samples for Vertex AI, including pipelines, metadata and more. Mainly with finance datasets.☆15Feb 7, 2026Updated 6 months ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- torch_remat fine-grained activation checkpointing API☆16Aug 19, 2026Updated last week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆456Jan 5, 2026Updated 7 months ago
- ☆155Aug 6, 2026Updated 3 weeks ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆66Aug 2, 2022Updated 4 years ago
- Conveniently export torch.compile compiled products into self-contained Python files☆34Jun 5, 2026Updated 2 months ago
- Multidisciplinary Research and Education on Big Data + High-Performance Computing + Atmospheric Sciences at UMBC☆15Jul 13, 2022Updated 4 years ago
- Developer blog for PyTorch☆20Updated this week
- Training NVIDIA NeMo Megatron Large Language Model (LLM) using NeMo Framework on Google Kubernetes Engine☆16Apr 28, 2025Updated last year
- ☆27Aug 24, 2026Updated last week
- Panorama and HDR photography on the Android Mobile platform using OpenCV libraries.☆17Aug 20, 2014Updated 12 years ago
- Prototype routines for GPU quantization written using PyTorch.☆21Apr 15, 2026Updated 4 months ago
- Fast, efficient code to pull non-null categorical data out, encode it and impute nulls with KNN Impute from fancyimpute library☆17Dec 8, 2019Updated 6 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ☆591Jul 11, 2024Updated 2 years ago
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆156Aug 17, 2026Updated 2 weeks ago
- Distributed Programming Abstractions☆16Dec 5, 2018Updated 7 years ago
- ☆42Dec 10, 2024Updated last year
- ☆11Apr 5, 2021Updated 5 years ago
- ☆20Mar 16, 2026Updated 5 months ago
- High-performance large-scale embedding acceleration for JAX on Google TPU SparseCores.☆37Updated this week