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
  ☆32Updated 4 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆57Updated 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
• danijar / ninjax
  General Modules for JAX
  ☆64Updated this week
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 4 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆133Updated 4 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆24Updated 8 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated last year
• yixiaoer / mistral-jax
  JAX implementation of the Mistral 7b v0.1 model
  ☆13Updated 11 months ago
• Reytuag / transformerXL_PPO_JAX
  ☆73Updated 4 months ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 3 months ago
• instadeepai / sebulba
  🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
  ☆56Updated last year
• keraJLi / synthetic-gymnax
  ☆20Updated 8 months ago
• google-deepmind / nao_top10
  ☆18Updated 2 years ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated 7 months ago
• DramaCow / jaxued
  ☆72Updated 6 months ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆48Updated last year
• chongyi-zheng / td_infonce
  ☆27Updated last year
• danijar / elements
  Building blocks for productive research
  ☆51Updated last month
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• htdt / lwm
  Latent World Models For Intrinsically Motivated Exploration | Official repository
  ☆22Updated 3 years ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 8 months ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆55Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆58Updated 3 years ago
• FLAIROx / cultural-accumulation
  ☆13Updated 7 months ago
• amacrutherford / sampling-for-learnability
  Official codebase for "Sampling For Learnability", published at NeurIPS 2024
  ☆13Updated last month
• CarperAI / Algorithm-Distillation-RLHF
  ☆34Updated 2 years ago
• MichaelTMatthews / Craftax_Baselines
  ☆18Updated last month