Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
☆64May 5, 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:
- This repository contains example code to build models on TPUs☆30Feb 17, 2023Updated 3 years ago
- ☆16Mar 13, 2025Updated last year
- ☆18Feb 18, 2026Updated 3 months ago
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆82Dec 18, 2025Updated 5 months ago
- ☆196May 4, 2026Updated 2 weeks 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.
- 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
- ☆30May 5, 2026Updated 2 weeks ago
- torchprime is a reference model implementation for PyTorch on TPU.☆47Mar 3, 2026Updated 2 months ago
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,779May 7, 2026Updated last week
- PyTorch centric eager mode debugger☆48Dec 16, 2024Updated last year
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆551May 8, 2026Updated last week
- ☆13May 9, 2026Updated last week
- ☆21Mar 3, 2025Updated last year
- 🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.☆17Jun 5, 2025Updated 11 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Google TPU optimizations for transformers models☆138Jan 23, 2026Updated 3 months ago
- A multi-platform experimentation framework written in python.☆66Updated this week
- Fast and easy distributed model training examples.☆12Nov 26, 2024Updated last year
- ☆31Updated this week
- This repository contains code samples for Vertex AI, including pipelines, metadata and more. Mainly with finance datasets.☆15Feb 7, 2026Updated 3 months ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆437Jan 5, 2026Updated 4 months ago
- ☆66Aug 2, 2022Updated 3 years ago
- ☆151May 6, 2026Updated 2 weeks ago
- 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.
- Automatically load data from Google Cloud Storage files into Big Query tables☆11Dec 30, 2022Updated 3 years ago
- Panorama and HDR photography on the Android Mobile platform using OpenCV libraries.☆17Aug 20, 2014Updated 11 years ago
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆79Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆332Updated this week
- Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.☆132Updated this week
- ☆25May 4, 2026Updated 2 weeks ago
- ☆20Feb 17, 2021Updated 5 years ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆507Updated this week
- 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
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆580Jul 11, 2024Updated last year
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆26Apr 20, 2023Updated 3 years ago
- PyTorch bindings for CUTLASS grouped GEMM.☆151May 29, 2025Updated 11 months ago
- This library has moved to https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-containeranalysis☆23Oct 31, 2023Updated 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…☆155May 6, 2026Updated 2 weeks ago
- ☆11Apr 5, 2021Updated 5 years ago
- Tensorflow2 training code with jit compiling on multi-GPU.☆17Jan 28, 2021Updated 5 years ago