danijar / teleport
Efficiently send large arrays across machines
☆16Updated 9 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 3 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated last week
- GPT implementation in Flax☆18Updated 3 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- Causal Analysis of Agent Behavior for AI Safety☆18Updated last year
- ☆31Updated 2 years ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated last year
- High quality implementations of imitation and inverse reinforcement learning algorithms☆15Updated last month
- flexible meta-learning in jax☆13Updated last year
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- Fast reinforcement learning 💨☆24Updated last month
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆22Updated last year
- Code for the paper "Function-Space Learning Rates"☆20Updated 3 weeks ago
- PyTorch Package For Quasimetric Learning☆42Updated 6 months ago
- ☆23Updated 3 years ago
- ☆42Updated 3 years ago
- Jax like function transformation engine but micro, microjax☆31Updated 6 months ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆27Updated last year
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆30Updated 5 months ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 3 months ago
- General Modules for JAX☆64Updated last month
- Repo to reproduce the First-Explore paper results☆37Updated 4 months ago
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)☆20Updated 8 months ago
- LMAct: A Benchmark for In-Context Imitation Learning with Long Multimodal Demonstrations☆16Updated 5 months ago
- ☆31Updated last year
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆35Updated 7 months ago
- Official Implementation of SFM and the baselines in Jax.☆18Updated 6 months ago
- Generalised UDRL☆37Updated 3 years ago