Keen-Technologies / physical_atariLinks
Platform for evaluating reinforcement learning (RL) algorithms on a physical Atari system.
☆142Updated 5 months ago
Alternatives and similar repositories for physical_atari
Users that are interested in physical_atari are comparing it to the libraries listed below
Sorting:
- ☆179Updated 2 months ago
- ☆61Updated 6 months ago
- seqax = sequence modeling + JAX☆170Updated 6 months ago
- ☆289Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆129Updated 3 weeks ago
- SIMD quantization kernels☆94Updated 4 months ago
- A 3D video game environment and benchmark designed from scratch for reinforcement learning research☆190Updated 2 years ago
- Cost aware hyperparameter tuning algorithm☆179Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆306Updated last year
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆73Updated last year
- ☆135Updated last month
- Simple Transformer in Jax☆142Updated last year
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆369Updated 7 months ago
- commaVQ is a dataset of compressed driving video☆347Updated 3 months ago
- Can you design a controller to steer a simulated car?☆342Updated 6 months ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆669Updated 5 months ago
- An implementation of delta-iris in tinygrad☆72Updated last year
- Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind☆406Updated 7 months ago
- MoE training for Me and You and maybe other people☆335Updated last month
- ☆214Updated last month
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.☆115Updated last month
- Async RL Training at Scale☆1,044Updated this week
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆375Updated last year
- Efficient baselines for autocurricula in JAX.☆206Updated last year
- Can RL solve simple problems?☆54Updated 2 years ago
- Dion optimizer algorithm☆424Updated 3 weeks ago
- Minimal yet performant LLM examples in pure JAX☆236Updated 3 weeks ago
- Solve puzzles. Learn CUDA.☆63Updated 2 years ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆63Updated 2 years ago
- Simple MPI implementation for prototyping or learning☆300Updated 6 months ago