danijar / teleportLinks
Efficiently send large arrays across machines
☆16Updated 10 months ago
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☆22Updated 4 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- Causal Analysis of Agent Behavior for AI Safety☆18Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated last week
- Fast reinforcement learning 💨☆24Updated 2 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated last year
- ☆31Updated 2 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- Fast and reliable distributed systems in Python☆28Updated last month
- Generalised UDRL☆37Updated 3 years ago
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆28Updated 7 months ago
- ☆42Updated 3 years ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆28Updated last year
- ☆23Updated 3 years ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- ☆15Updated 2 years ago
- Implementation of CASCADE in Learning General World Models in a Handful of Reward-Free Deployments (NeurIPS 22).☆29Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆42Updated 7 months ago
- Repo to reproduce the First-Explore paper results☆37Updated 5 months ago
- General Modules for JAX☆66Updated last month
- ☆13Updated 10 months ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- Jax like function transformation engine but micro, microjax☆32Updated 7 months ago
- ☆20Updated last year
- Learn online intrinsic rewards from LLM feedback☆37Updated 5 months ago
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆22Updated last year
- ☆32Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated 4 months ago
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago