Alternatives and similar repositories for mctx:

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

• luchris429 / purejaxrl
  Really Fast End-to-End Jax RL Implementations
  ☆859Updated 7 months ago
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆3,013Updated this week
• google-deepmind / rlax
  ☆1,309Updated last week
• google / evojax
  ☆897Updated 9 months ago
• eloialonso / iris
  Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.
  ☆835Updated 6 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,868Updated this week
• instadeepai / jumanji
  🕹️ A diverse suite of scalable reinforcement learning environments in JAX
  ☆719Updated 3 weeks ago
• YeWR / EfficientZero
  Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.
  ☆892Updated last year
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,317Updated last week
• nnaisense / evotorch
  Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.
  ☆1,052Updated last week
• sotetsuk / pgx
  ♟️ Vectorized RL game environments in JAX
  ☆465Updated last month
• n2cholas / awesome-jax
  JAX - A curated list of resources https://github.com/google/jax
  ☆1,780Updated last month
• google / vizier
  Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
  ☆1,552Updated last week
• google-deepmind / chex
  ☆845Updated 2 weeks ago
• RobertTLange / gymnax
  RL Environments in JAX 🌍
  ☆742Updated 9 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,758Updated 2 weeks ago
• google / learned_optimization
  ☆776Updated last week
• alex-petrenko / sample-factory
  High throughput synchronous and asynchronous reinforcement learning
  ☆888Updated 3 months ago
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,375Updated 3 weeks ago
• microsoft / mup
  maximal update parametrization (µP)
  ☆1,497Updated 9 months ago
• facebookresearch / Pearl
  A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.
  ☆2,807Updated this week
• google / brax
  Massively parallel rigidbody physics simulation on accelerator hardware.
  ☆2,624Updated last week
• pytorch / rl
  A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
  ☆2,685Updated this week
• pytorch-labs / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆545Updated 5 months ago
• google-deepmind / tree
  tree is a library for working with nested data structures
  ☆979Updated 2 months ago
• google-deepmind / alphadev
  ☆709Updated last year
• google-deepmind / xmanager
  A platform for managing machine learning experiments
  ☆846Updated last month
• google-research / rliable
  [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.
  ☆825Updated 8 months ago
• AgileRL / AgileRL
  Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.
  ☆737Updated this week
• google-deepmind / jraph
  A Graph Neural Network Library in Jax
  ☆1,423Updated last year