TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
☆20Sep 24, 2025Updated last year
Alternatives and similar repositories for tpu_pod_commander
Users that are interested in tpu_pod_commander are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Minimal but scalable implementation of large language models in JAX☆34Nov 28, 2025Updated 11 months ago
- Conservative Q learning in Jax☆58Feb 7, 2023Updated 3 years ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆27Sep 29, 2024Updated 2 years ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated last year
- 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.
- Make-A-Video Latent Diffusion Model☆19Nov 15, 2023Updated 2 years ago
- Quantifying degeneracy in toy models☆11Sep 6, 2024Updated 2 years ago
- Environment codebase for ICRA 2020 paper "Towards Practical Multi-object Manipulation using Relational Reinforcement Learning"☆14Jul 22, 2020Updated 6 years ago
- ☆11Jul 29, 2023Updated 3 years ago
- An implementation of PSGD Kron in JAX for distributed training in JAX or Flax☆10Nov 6, 2025Updated last year
- The NYU Systems Seminar☆25Feb 26, 2024Updated 2 years ago
- [NeurIPS'25] Towards Interpretability Without Sacrifice: Faithful Dense Layer Decomposition with Mixture of Decoders☆16May 28, 2025Updated last year
- ICRL 2020☆20Feb 18, 2020Updated 6 years ago
- JAX-compatible, simplex method-based linear program solver☆12May 24, 2026Updated 5 months ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆18Jul 13, 2022Updated 4 years ago
- Gym-like environment to interact with manipulator robots☆44Mar 7, 2025Updated last year
- If it quacks like a tensor...☆59Nov 13, 2024Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 3 years ago
- ☆15May 11, 2025Updated last year
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Jul 28, 2023Updated 3 years ago
- [EMNLP 2024] Is C4 Dataset Optimal for Pruning? An Investigation of Calibration Data for LLM Pruning☆10Oct 11, 2024Updated 2 years ago
- Fast, simple, cryptographically strong random numbers in C++. Experimental.☆19Dec 12, 2013Updated 12 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.
- SeeQ: subtask-aware PaliGemma Q-critics for best-of-N steering of pi0.5 policies (JAX)☆19Sep 29, 2026Updated last month
- Train very large language models in Jax.☆209Oct 21, 2023Updated 3 years ago
- JAX Synergistic Memory Inspector☆187Jul 16, 2024Updated 2 years ago
- ☆35Jan 29, 2023Updated 3 years ago
- coloring terminal text with intensities (used for plotting probability, entropy with tokens)☆12Oct 11, 2024Updated 2 years ago
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated 2 years ago
- Fork of Triton repository for OpenXLA uses of the Triton language and compiler☆18Oct 6, 2026Updated last month
- A simple library for scaling up JAX programs☆149Nov 4, 2025Updated last year
- Everything you want to know about Google Cloud TPU☆573Jul 16, 2024Updated 2 years 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.
- ☆15May 11, 2026Updated 5 months ago
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- A reliable leaderboard algorithm for machine learning competitions☆17May 19, 2015Updated 11 years ago
- ☆14May 16, 2024Updated 2 years ago
- ☆10Jul 31, 2025Updated last year
- [NeurIPS2022] This is the official implementation of the paper "Expediting Large-Scale Vision Transformer for Dense Prediction without Fi…☆10Mar 23, 2023Updated 3 years ago
- ☆13Jan 16, 2019Updated 7 years ago