danijar / teleportLinks
Efficiently send large arrays across machines
☆16Updated last year
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 7 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 2 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- Fast reinforcement learning 💨☆26Updated last month
- PyTorch Package For Quasimetric Learning☆42Updated 10 months ago
- Fast and reliable distributed systems in Python☆28Updated 4 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- General Modules for JAX☆67Updated 4 months ago
- ☆32Updated 9 months ago
- Clean RL implementation using MLX☆32Updated last year
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- Repo to reproduce the First-Explore paper results☆38Updated 8 months ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- Causal Analysis of Agent Behavior for AI Safety☆18Updated 2 years ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆62Updated 10 months ago
- Building blocks for productive research☆59Updated last month
- Generalised UDRL☆37Updated 3 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆31Updated 2 years ago
- Jax like function transformation engine but micro, microjax☆33Updated 10 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 11 months ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆18Updated last year
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆107Updated 11 months ago
- Corax: Core RL in JAX☆38Updated last year
- ☆15Updated 2 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Updated 3 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆87Updated last year
- Reinforcement learning in pure JAX.☆13Updated 6 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated 2 years ago
- Accelerated replay buffers in JAX☆43Updated 2 years ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 5 years ago