Pinned Loading

1. CognitiveAISystems/MIKASA-Robo CognitiveAISystems/MIKASA-Robo Public

   🔥 [ICLR 2026] Benchmark for robotic tabletop manipulation memory-intensive tasks

   Jupyter Notebook 50 8

2. CognitiveAISystems/RATE CognitiveAISystems/RATE Public

   🔥 [ICLR 2026] Official implementation of Recurrent Action Transformer with Memory, an offline RL agent with memory mechanisms. https://sites.google.com/view/rate-model/

   Python 19 2

3. awesome-memory-vla awesome-memory-vla Public

   🧠 Awesome Memory-VLA: A curated list of Visual-Language-Action models with memory

   32

4. lucidrains/recurrent-memory-transformer-pytorch lucidrains/recurrent-memory-transformer-pytorch Public

   Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch

   Python 422 20

5. CognitiveAISystems/kage-bench CognitiveAISystems/kage-bench Public

   KAGE-Bench: pure JAX 2D platformer RL benchmark for visual OOD generalization. Massively-parallel (vmap/JIT) RGB env with YAML-configurable visuals/physics, plus PPO-CNN (Flax) training scripts.

   Python 11

6. dmitrySorokin/aimasters_2025 dmitrySorokin/aimasters_2025 Public

   Jupyter Notebook 9