Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
☆64Jul 22, 2026Updated 3 months 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:
- 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
- ☆16Mar 13, 2025Updated last year
- This repository contains example code to build models on TPUs☆30Feb 17, 2023Updated 3 years ago
- ☆200Oct 20, 2026Updated last week
- ☆19Feb 18, 2026Updated 8 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆84Dec 18, 2025Updated 10 months ago
- Inference code for LLaMA models☆20Apr 3, 2025Updated last year
- torchprime is a reference model implementation for PyTorch on TPU.☆49Mar 3, 2026Updated 7 months 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
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆560Oct 15, 2026Updated 2 weeks ago
- ☆13Jun 16, 2026Updated 4 months ago
- ☆33Oct 5, 2026Updated 3 weeks ago
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,807Oct 20, 2026Updated last week
- ☆21Mar 3, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆49May 5, 2026Updated 5 months ago
- "Not too complicated" training code for CIFAR-10 by PyTorch Lightning☆12Jun 5, 2022Updated 4 years ago
- Local and global SPMD type system supporting erasure☆10Apr 24, 2026Updated 6 months ago
- ☆12May 10, 2023Updated 3 years ago
- Moving to vllm-project/vllm-jax. TPU inference for vLLM, with unified JAX and PyTorch support.☆471Updated this week
- Google TPU optimizations for transformers models☆137Jan 23, 2026Updated 9 months ago
- A tool for anonymizing webserver logs☆15Apr 23, 2017Updated 9 years ago
- ☆39Oct 12, 2026Updated 2 weeks ago
- Tokamax: A GPU and TPU kernel library.☆322Updated this week
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- A multi-platform experimentation framework written in python.☆72Updated this week
- Inference Llama 2 with a model compiled to native code by TorchInductor☆14Feb 8, 2024Updated 2 years ago
- K(ickstart)r for Typesafe projects☆11Oct 24, 2015Updated 11 years ago
- A sample pattern for running CI tests on Modal☆21Apr 12, 2025Updated last year
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- Automatically load data from Google Cloud Storage files into Big Query tables☆10Dec 30, 2022Updated 3 years ago
- Node.js wrapper of Merriam Webster Dictionary & Thesaurus Developer API☆10Feb 12, 2026Updated 8 months ago
- Pembrolizumab-like Antibody Hallucination using AlphaFold2☆16Nov 10, 2023Updated 2 years ago
- Conveniently export torch.compile compiled products into self-contained Python files☆35Jun 5, 2026Updated 4 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆66Aug 2, 2022Updated 4 years ago
- ☆156Oct 20, 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…☆472Jan 5, 2026Updated 9 months ago
- Tools and scripts for Google Compute Engine images.☆218Oct 23, 2026Updated last week
- Prototype routines for GPU quantization written using PyTorch.☆21Apr 15, 2026Updated 6 months ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆27Apr 20, 2023Updated 3 years ago
- torch_remat fine-grained activation checkpointing API☆23Updated this week