Alternatives and similar repositories for physical_atari

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

• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆168Updated 4 months ago
• google-deepmind / nanodo
  ☆285Updated last year
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆176Updated last year
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆357Updated 5 months ago
• meta-pytorch / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆657Updated 3 months ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆71Updated 11 months ago
• kvfrans / lmpo
  ☆128Updated 3 weeks ago
• iliao2345 / CompressARC
  ☆201Updated 3 months ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase.
  ☆116Updated last month
• jrosseruk / Torch2Jax-DeepSeek-R1-Distill-Qwen-1.5B
  Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.
  ☆26Updated 9 months ago
• NetHack-LE / nle
  The NetHack Learning Environment
  ☆96Updated last week
• drubinstein / pokemonred_puffer
  ☆176Updated last week
• PufferAI / PufferTank
  ☆56Updated 5 months ago
• Metta-AI / metta
  A reinforcement learning codebase focusing on the emergence of cooperation and alignment in multi-agent AI systems.
  ☆134Updated this week
• LeonGuertler / UnstableBaselines
  ☆108Updated last week
• AllanYangZhou / midGPT
  Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.
  ☆24Updated last year
• thinking-machines-lab / tinker
  Training API and CLI
  ☆253Updated last week
• microsoft / dion
  Dion optimizer algorithm
  ☆403Updated last week
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆201Updated last year
• Avalon-Benchmark / avalon
  A 3D video game environment and benchmark designed from scratch for reinforcement learning research
  ☆190Updated 2 years ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆106Updated 2 weeks ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆62Updated 2 years ago
• PrimeIntellect-ai / prime-rl
  Async RL Training at Scale
  ☆938Updated this week
• lucidrains / q-transformer
  Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind
  ☆405Updated 5 months ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆532Updated 4 months ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆140Updated last year
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 weeks ago
• PrimeIntellect-ai / pi-quant
  SIMD quantization kernels
  ☆93Updated 3 months ago
• kjaved0 / swifttd
  A fast and robust algorithm for temporal difference learning
  ☆21Updated last week
• balrog-ai / BALROG
  Benchmarking Agentic LLM and VLM Reasoning On Games
  ☆214Updated last week