young-geng / tpu_pod_commander
TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
☆20Updated 8 months ago
Alternatives and similar repositories for tpu_pod_commander:
Users that are interested in tpu_pod_commander are comparing it to the libraries listed below
- Minimal but scalable implementation of large language models in JAX☆32Updated 4 months ago
- If it quacks like a tensor...☆57Updated 3 months ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- General Modules for JAX☆64Updated this week
- PyTorch Package For Quasimetric Learning☆41Updated 4 months ago
- A simple library for scaling up JAX programs☆133Updated 4 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆24Updated 8 months ago
- flexible meta-learning in jax☆12Updated last year
- LoRA for arbitrary JAX models and functions☆135Updated last year
- JAX implementation of the Mistral 7b v0.1 model☆13Updated 11 months ago
- ☆73Updated 4 months ago
- ☆30Updated 3 months ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆56Updated last year
- ☆20Updated 8 months ago
- ☆18Updated 2 years ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 7 months ago
- ☆72Updated 6 months ago
- Scaling scaling laws with board games.☆48Updated last year
- ☆27Updated last year
- Building blocks for productive research☆51Updated last month
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Latent World Models For Intrinsically Motivated Exploration | Official repository☆22Updated 3 years ago
- Machine Learning eXperiment Utilities☆46Updated 8 months ago
- An implementation of MuZero in JAX.☆55Updated 2 years ago
- ☆58Updated 3 years ago
- ☆13Updated 7 months ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆13Updated last month
- ☆34Updated 2 years ago
- ☆18Updated last month