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 7 months ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆58Updated 7 months ago
• danijar / ninjax
  General Modules for JAX
  ☆65Updated 2 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆139Updated 7 months ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆139Updated last year
• kvfrans / splus
  ☆104Updated 2 weeks ago
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆20Updated 3 weeks ago
• radarFudan / mamba-minimal-jax
  ☆31Updated 7 months ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated last year
• 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
• danijar / elements
  Building blocks for productive research
  ☆59Updated 4 months ago
• google-deepmind / hierarchical_perceiver
  ☆27Updated last month
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆51Updated 6 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆14Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• okarthikb / DPO
  Implementation of Direct Preference Optimization
  ☆16Updated last year
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆41Updated 6 months ago
• danijar / portal
  Fast and reliable distributed systems in Python
  ☆28Updated 2 months ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated 11 months ago
• maxreciprocate / offline
  Offline RL experiments
  ☆15Updated 2 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 7 months ago
• chongyi-zheng / td_infonce
  Implementations of Temporal Difference InfoNCE (TD InfoNCE)
  ☆29Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• AllanYangZhou / universal_neural_functional
  ☆51Updated last year
• enajx / HyperNCA
  ☆39Updated 3 years ago
• yixiaoer / mistral-jax
  JAX implementation of the Mistral 7b v0.1 model
  ☆13Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• FLAIROx / cultural-accumulation
  ☆12Updated 11 months ago
• davisyoshida / haiku-mup
  A port of muP to JAX/Haiku
  ☆25Updated 2 years ago