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 7 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 6 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆137Updated 7 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• danijar / ninjax
  General Modules for JAX
  ☆66Updated 2 months ago
• radarFudan / mamba-minimal-jax
  ☆29Updated 6 months ago
• danijar / elements
  Building blocks for productive research
  ☆55Updated 4 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 11 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• google-deepmind / spectral_ssm
  ☆32Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆29Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• CarperAI / Algorithm-Distillation-RLHF
  ☆34Updated 2 years ago
• google-deepmind / nao_top10
  ☆18Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 7 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆14Updated last year
• ethansmith2000 / TransformerExperiments
  ☆19Updated 3 weeks ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆49Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• jacooba / hyper
  Code for the papers Hypernetworks in Meta-Reinforcement Learning (Beck et al., 2022) and Recurrent Hypernetworks are Surprisingly Strong …
  ☆14Updated 10 months ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆50Updated 6 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated last week
• chongyi-zheng / td_infonce
  Implementations of Temporal Difference InfoNCE (TD InfoNCE)
  ☆28Updated last year
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆37Updated 5 months ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆127Updated last year
• sustcsonglin / mamba-triton
  ☆47Updated last year
• GallagherCommaJack / modulax
  ☆17Updated 9 months ago