danijar / teleport
Efficiently send large arrays across machines
☆15Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for teleport
- ☆29Updated 2 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆29Updated 2 weeks ago
- GPT implementation in Flax☆18Updated 2 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- ☆17Updated 5 months ago
- ☆28Updated 7 months ago
- ☆15Updated last month
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆41Updated last year
- ☆29Updated 2 months ago
- Causal Analysis of Agent Behavior for AI Safety☆17Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated last month
- Implementation of BC-IRL and other IRL baselines☆25Updated last year
- This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".☆20Updated 11 months ago
- Write simple games in Numpy!☆12Updated 2 years ago
- ☆15Updated 2 years ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated last year
- PyTorch Package For Quasimetric Learning☆42Updated 3 weeks ago
- Fast dataset format and loader☆21Updated last month
- Generalised UDRL☆37Updated 2 years ago
- Building blocks for productive research☆45Updated last week
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆26Updated 2 years ago
- ☆17Updated last year
- MACTA: A Multi-agent Reinforcement Learning Approach for Cache Timing Attacks and Detection☆45Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated last month
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Latent World Models For Intrinsically Motivated Exploration | Official repository☆21Updated 3 years ago
- A library for simplifying fine tuning with multi gpu setups in the Huggingface ecosystem.☆15Updated 3 weeks ago
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 3 years ago
- ☆24Updated 7 months ago