Alternatives and similar repositories for metalgpu

Users that are interested in metalgpu are comparing it to the libraries listed below

• jax-ml / jax-llm-examples
  ☆135Updated this week
• lowrollr / turbozero
  fast + parallel AlphaZero in JAX
  ☆97Updated 6 months ago
• kvfrans / splus
  ☆111Updated last month
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆116Updated this week
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆187Updated this week
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆162Updated last year
• zach-lawless / gym-wordle
  Gym environment for playing Wordle with RL agents
  ☆39Updated 3 years ago
• bwfbowen / muax
  A project that provides help for using DeepMind's mctx on gym-style environments.
  ☆60Updated 8 months ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆323Updated this week
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆21Updated last year
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆190Updated 10 months ago
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆230Updated last month
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆404Updated this week
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆141Updated last month
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆239Updated 3 months ago
• CLAIRE-Labo / no-representation-no-trust
  Codebase to fully reproduce the results of "No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPO" (M…
  ☆27Updated 8 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆281Updated last week
• PufferAI / pokegym
  Gymnasium environment for Pokemon Red
  ☆38Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆60Updated 6 months ago
• google-deepmind / nanodo
  ☆274Updated last year
• FLAIROx / Kinetix
  Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
  ☆195Updated last week
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆206Updated 5 months ago
• PufferAI / PufferTank
  ☆41Updated last week
• GAIGResearch / Tribes
  Tribes is an open source implementation for AI research of the game The Battle of Polytopia
  ☆74Updated 2 years ago
• mikelma / craftium
  A framework for creating rich, 3D, Minecraft-like single and multi-agent environments for AI research. (Accepted at ICML 2025).
  ☆113Updated last week
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆256Updated 2 weeks ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆206Updated 2 months ago
• google-research / dataclass_array
  Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
  ☆50Updated 10 months ago
• morgangiraud / leniax
  Leniax is a search and rendering engine for Lenia.
  ☆27Updated last year
• HumanCompatibleAI / leela-interp
  Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"
  ☆24Updated last year