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
  ☆35Updated 2 weeks ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 8 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆140Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆139Updated 8 months ago
• radarFudan / mamba-minimal-jax
  ☆31Updated 7 months ago
• danijar / ninjax
  General Modules for JAX
  ☆66Updated 3 months ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆204Updated last year
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 3 weeks ago
• kvfrans / splus
  ☆111Updated last month
• google-deepmind / nao_top10
  ☆18Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆103Updated 2 years ago
• danijar / elements
  Building blocks for productive research
  ☆59Updated 5 months ago
• sholtodouglas / scalingExperiments
  ☆61Updated 3 years ago
• jacooba / hyper
  Code for the papers Hypernetworks in Meta-Reinforcement Learning (Beck et al., 2022) and Recurrent Hypernetworks are Surprisingly Strong …
  ☆15Updated 11 months ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆49Updated 2 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆98Updated 3 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆30Updated last year
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆129Updated last year
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆11Updated last year
• smonsays / metax
  flexible meta-learning in jax
  ☆14Updated last year
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆53Updated 7 months ago
• AllanYangZhou / universal_neural_functional
  ☆51Updated last year
• AllanYangZhou / midGPT
  Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.
  ☆24Updated 9 months ago
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆41Updated 7 months ago
• amacrutherford / sampling-for-learnability
  Official codebase for "Sampling For Learnability", published at NeurIPS 2024
  ☆16Updated last month
• Reytuag / transformerXL_PPO_JAX
  ☆81Updated 8 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 6 months ago