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 3 weeks ago
- GPT implementation in Flax☆18Updated 4 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated 2 years ago
- Causal Analysis of Agent Behavior for AI Safety☆19Updated 2 years ago
- General Modules for JAX☆72Updated 4 months ago
- Generalised UDRL☆37Updated 3 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 7 months ago
- Reinforcement learning in pure JAX.☆13Updated last month
- An implementation of MuZero in JAX.☆57Updated 3 years ago
- Repo to reproduce the First-Explore paper results☆39Updated last year
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated 2 years ago
- Corax: Core RL in JAX☆38Updated last year
- Fast reinforcement learning 💨☆28Updated 6 months ago
- PyTorch Package For Quasimetric Learning☆45Updated last year
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 3 years ago
- Fast and reliable distributed systems in Python☆33Updated 2 weeks ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Updated 2 years ago
- flexible meta-learning in jax☆16Updated 2 years ago
- ☆23Updated 4 years ago
- Clean RL implementation using MLX☆34Updated last year
- Accelerated replay buffers in JAX☆46Updated 3 years ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆66Updated last year
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 4 years ago
- ☆42Updated 3 years ago
- Building blocks for productive research☆67Updated 2 weeks ago
- Understanding RL vision Distill article☆25Updated 2 years ago
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆47Updated 4 years ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆23Updated 5 years ago
- GBRL-based Actor-Critic algorithms implemented in stable-baselines3☆43Updated this week
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆29Updated 3 years ago