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 5 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 5 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 5 months ago
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆35Updated 4 months ago
• danijar / elements
  Building blocks for productive research
  ☆52Updated 2 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆28Updated 10 months ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 5 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆13Updated last year
• danijar / ninjax
  General Modules for JAX
  ☆64Updated 2 weeks ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆50Updated 4 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆31Updated 3 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 10 months ago
• haoliuhl / tux
  Tools and Utils for Experiments (TUX)
  ☆16Updated 2 months ago
• instadeepai / sebulba
  🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
  ☆57Updated last year
• google-deepmind / nao_top10
  ☆18Updated 2 years ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆61Updated 3 months ago
• AllanYangZhou / universal_neural_functional
  ☆49Updated last year
• AllanYangZhou / midGPT
  Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.
  ☆24Updated 6 months ago
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆19Updated last week
• DramaCow / jaxued
  ☆76Updated last month
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated 9 months ago
• google-deepmind / hierarchical_perceiver
  ☆26Updated 9 months ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 5 months ago
• danijar / portal
  Fast and reliable distributed systems in Python
  ☆25Updated this week
• astanic / crafter-ood
  ☆20Updated 2 years ago
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆27Updated 6 months ago