Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
☆64Jul 22, 2026Updated 2 weeks ago
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
- A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across differe…☆65Updated this week
- ☆200Jul 22, 2026Updated 2 weeks ago
- ☆19Feb 18, 2026Updated 5 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆86Dec 18, 2025Updated 7 months ago
- Inference code for LLaMA models☆20Apr 3, 2025Updated last year
- 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
- ☆32Jul 9, 2026Updated last month
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,798May 27, 2026Updated 2 months ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆558Updated this week
- PyTorch centric eager mode debugger☆46Dec 16, 2024Updated last year
- ☆21Mar 3, 2025Updated last year
- 🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.☆17Jun 5, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- "Not too complicated" training code for CIFAR-10 by PyTorch Lightning☆12Jun 5, 2022Updated 4 years ago
- Google TPU optimizations for transformers models☆135Jan 23, 2026Updated 6 months ago
- Tokamax: A GPU and TPU kernel library.☆259Updated this week
- This is an SDK for Google Cloud Explainable AI service. Explainable AI SDK helps users build explanation metadata for their models and vi…☆24Jan 25, 2022Updated 4 years ago
- ☆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
- A multi-platform experimentation framework written in python.☆68Updated this week
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- torch_remat fine-grained activation checkpointing API☆15Jul 28, 2026Updated last week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆451Jan 5, 2026Updated 7 months ago
- ☆154Updated this week
- ☆66Aug 2, 2022Updated 4 years ago
- Conveniently export torch.compile compiled products into self-contained Python files☆35Jun 5, 2026Updated 2 months ago
- Developer blog for PyTorch☆20Jul 28, 2026Updated last week
- ☆26Updated this week
- Prototype routines for GPU quantization written using PyTorch.☆21Apr 15, 2026Updated 3 months ago
- Fast, efficient code to pull non-null categorical data out, encode it and impute nulls with KNN Impute from fancyimpute library