danijar / teleport
Efficiently send large arrays across machines
☆15Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for teleport
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆29Updated last week
- Jax like function transformation engine but micro, microjax☆26Updated 2 weeks ago
- GPT implementation in Flax☆18Updated 2 years ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆41Updated last year
- MACTA: A Multi-agent Reinforcement Learning Approach for Cache Timing Attacks and Detection☆45Updated last year
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)☆18Updated 2 months ago
- Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models☆45Updated 5 months ago
- ☆29Updated 2 years ago
- Causal Analysis of Agent Behavior for AI Safety☆17Updated last year
- AgentHive provides the primitives and helpers for a seamless usage of robohive within TorchRL.☆32Updated 10 months ago
- ☆17Updated last year
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks☆20Updated last week
- ☆29Updated last month
- Repo to reproduce the First-Explore paper results☆36Updated last week
- PyTorch Package For Quasimetric Learning☆42Updated last week
- Clean RL implementation using MLX☆26Updated 8 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated last month
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- ☆22Updated 3 years ago
- ☆17Updated 4 months ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated last year
- Implementation of BC-IRL and other IRL baselines☆25Updated last year
- General Modules for JAX☆58Updated 3 months ago
- ☆15Updated 2 years ago
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆22Updated 2 weeks ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆26Updated 9 months ago
- Code for the paper "Understanding RL Vision"☆43Updated last year
- Clockwork VAEs in JAX/Flax☆31Updated 3 years ago