Alternatives and similar repositories for tpu_pod_commander

Users that are interested in tpu_pod_commander are comparing it to the libraries listed below

• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆34Updated 6 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 6 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 6 months ago
• danijar / ninjax
  General Modules for JAX
  ☆65Updated last month
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆29Updated 11 months ago
• danijar / elements
  Building blocks for productive research
  ☆53Updated 3 months ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 5 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆13Updated last year
• haoliuhl / tux
  Tools and Utils for Experiments (TUX)
  ☆16Updated 3 months ago
• astanic / crafter-ood
  ☆20Updated 2 years ago
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆37Updated 5 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 11 months ago
• TheDuckAI / prm
  ☆12Updated 4 months ago
• maxreciprocate / offline
  Offline RL experiments
  ☆14Updated 2 years ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 4 months ago
• google-deepmind / nao_top10
  ☆18Updated 2 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 6 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆20Updated last month
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆50Updated 5 months ago
• rystrauss / dopamax
  Reinforcement learning in pure JAX.
  ☆12Updated 2 months ago
• yixiaoer / mistral-jax
  JAX implementation of the Mistral 7b v0.1 model
  ☆13Updated last year
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated 10 months ago
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆28Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated this week
• FLAIROx / cultural-accumulation
  ☆13Updated 10 months ago
• google-deepmind / spectral_ssm
  ☆31Updated last year
• MichaelTMatthews / Craftax_Baselines
  ☆19Updated this week