Alternatives and similar repositories for wandb-workspaces

Users that are interested in wandb-workspaces are comparing it to the libraries listed below

• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆205Updated last week
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆50Updated last year
• yobibyte / compressor
  Because we don't have enough time to read everything
  ☆89Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated 2 months ago
• oripress / AlgoTune
  AlgoTune is a NeurIPS 2025 benchmark made up of 154 math, physics, and computer science problems. The goal is write code that solves each…
  ☆81Updated this week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆309Updated 2 months ago
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆186Updated this week
• google / torchax
  torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…
  ☆162Updated this week
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆34Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆110Updated 2 months ago
• mila-iqia / milabench
  Repository of machine learning benchmarks
  ☆49Updated last month
• cgarciae / einop
  ☆60Updated 3 years ago
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆456Updated 9 months ago
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆70Updated 3 months ago
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆177Updated last year
• google-deepmind / tf2jax
  ☆118Updated last month
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆155Updated 2 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆118Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 4 months ago
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆175Updated this week
• saforem2 / ambivalent
  Minimal, beautiful (+ highly-customizable) styles for Matplotlib.
  ☆19Updated last month
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• facebookresearch / oni
  Learn online intrinsic rewards from LLM feedback
  ☆45Updated last year
• openai / chz
  ☆213Updated last month
• Artur-Galstyan / statedict2pytree
  ☆44Updated 2 months ago
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 7 months ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆297Updated last year
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• Farama-Foundation / Jumpy
  On-the-fly conversions between Jax and NumPy tensors
  ☆57Updated 2 years ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆136Updated 3 weeks ago