Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
☆64May 5, 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:
- 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
- ☆197May 4, 2026Updated last month
- 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.
- ☆32Jun 3, 2024Updated 2 years ago
- 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
- ☆31Jun 2, 2026Updated last week
- torchprime is a reference model implementation for PyTorch on TPU.☆47Mar 3, 2026Updated 3 months ago
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,786May 27, 2026Updated 2 weeks ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆554Updated this week
- ☆13May 9, 2026Updated last month
- ☆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 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.
- Google TPU optimizations for transformers models☆137Jan 23, 2026Updated 4 months ago
- "Not too complicated" training code for CIFAR-10 by PyTorch Lightning☆12Jun 5, 2022Updated 4 years ago
- Fast and easy distributed model training examples.☆12Nov 26, 2024Updated last year
- ☆12May 10, 2023Updated 3 years ago
- ☆22Dec 14, 2021Updated 4 years ago
- Inference code for LLaMA models☆21Apr 3, 2025Updated last year
- ☆32Updated this week
- This repository contains code samples for Vertex AI, including pipelines, metadata and more. Mainly with finance datasets.☆15Feb 7, 2026Updated 4 months ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years 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.
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆443Jan 5, 2026Updated 5 months ago
- ☆153May 29, 2026Updated last week
- Multidisciplinary Research and Education on Big Data + High-Performance Computing + Atmospheric Sciences at UMBC☆16Jul 13, 2022Updated 3 years ago
- Automatically load data from Google Cloud Storage files into Big Query tables☆11Dec 30, 2022Updated 3 years ago
- GPU Stress Test is a tool to stress the compute engine of NVIDIA Tesla GPU’s by running a BLAS matrix multiply using different data types…☆123Jul 8, 2025Updated 11 months ago
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆83Jun 2, 2026Updated last week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆349Updated this week
- Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.☆133Updated this week
- ☆20Feb 17, 2021Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆582Jul 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.☆152May 29, 2025Updated last year
- 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 last month
- URL downloader supporting checkpointing and continuous checksumming.☆19Nov 29, 2023Updated 2 years ago
- ☆42Dec 10, 2024Updated last year
- This library has moved to https://github.com/googleapis/google-cloud-python/tree/main/packages/google-cloud-bigquery-datatransfer☆84Sep 29, 2023Updated 2 years ago