danijar / teleport
Efficiently send large arrays across machines
☆15Updated 8 months ago
Alternatives and similar repositories for teleport:
Users that are interested in teleport are comparing it to the libraries listed below
- Fast dataset format and loader☆22Updated 2 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 3 months ago
- Causal Analysis of Agent Behavior for AI Safety☆17Updated last year
- 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
- GPT implementation in Flax☆18Updated 3 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- Code for the paper Task Agnostic Morphology Evolution.☆20Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆41Updated 4 months ago
- Learn online intrinsic rewards from LLM feedback☆35Updated 3 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 6 months ago
- Fast reinforcement learning 💨☆24Updated last week
- Jax like function transformation engine but micro, microjax☆30Updated 5 months ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated last year
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)☆20Updated 7 months ago
- Implementation of BC-IRL and other IRL baselines☆27Updated last year
- ☆15Updated 2 years ago
- ☆42Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Reinforcement learning in pure JAX.☆11Updated last month
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆18Updated 3 years ago
- ☆30Updated 4 months ago
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆22Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- Official Implementation of SFM and the baselines in Jax.☆15Updated 4 months ago
- Latent World Models For Intrinsically Motivated Exploration | Official repository☆22Updated 3 years ago
- MACTA: A Multi-agent Reinforcement Learning Approach for Cache Timing Attacks and Detection☆46Updated last year
- flexible meta-learning in jax☆12Updated last year
- code for paper "Accessing higher dimensions for unsupervised word translation"☆21Updated last year