danijar / teleportLinks
Efficiently send large arrays across machines
☆17Updated last year
Alternatives and similar repositories for teleport
Users that are interested in teleport are comparing it to the libraries listed below
Sorting:
- Fast dataset format and loader☆23Updated last week
- GPT implementation in Flax☆18Updated 4 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated 2 years ago
- Fast reinforcement learning 💨☆28Updated 5 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 7 months ago
- PyTorch Package For Quasimetric Learning☆45Updated last year
- Repo to reproduce the First-Explore paper results☆38Updated last year
- ☆35Updated last year
- Building blocks for productive research☆67Updated this week
- Corax: Core RL in JAX☆38Updated last year
- Reinforcement learning in pure JAX.☆13Updated 2 weeks ago
- An implementation of MuZero in JAX.☆58Updated 3 years ago
- Fast and reliable distributed systems in Python☆33Updated 8 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆66Updated last year
- Generative cellular automaton-like learning environments for RL.☆20Updated 11 months ago
- General Modules for JAX☆72Updated 3 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆116Updated last year
- Generalised UDRL☆37Updated 3 years ago
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆25Updated 2 years ago
- ☆16Updated 3 years ago
- Clean RL implementation using MLX☆34Updated last year
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated 2 years ago
- A simple, performant and scalable JAX-based world modeling codebase.☆119Updated 2 months ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 3 years ago
- Accelerated replay buffers in JAX☆46Updated 3 years ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆79Updated last year
- MACTA: A Multi-agent Reinforcement Learning Approach for Cache Timing Attacks and Detection☆46Updated 2 years ago
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks☆36Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Updated 2 years ago