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 8 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- Fast reinforcement learning 💨☆26Updated 2 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 3 months ago
- Fast and reliable distributed systems in Python☆28Updated 5 months ago
- Generalised UDRL☆37Updated 3 years ago
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆43Updated 10 months ago
- ☆34Updated 10 months ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- Corax: Core RL in JAX☆38Updated last year
- Repo to reproduce the First-Explore paper results☆38Updated 9 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated 2 years ago
- Jax like function transformation engine but micro, microjax☆33Updated 11 months ago
- General Modules for JAX☆67Updated 2 weeks ago
- Clean RL implementation using MLX☆33Updated last year
- Building blocks for productive research☆61Updated last month
- Reinforcement learning in pure JAX.☆13Updated 7 months ago
- ☆15Updated 3 years ago
- Understanding RL vision Distill article☆24Updated 2 years ago
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks☆35Updated 10 months ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆22Updated 4 years ago
- ☆23Updated 3 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆31Updated 2 years ago
- Accelerated replay buffers in JAX☆43Updated 3 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Updated 3 years ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆27Updated last year
- Causal Analysis of Agent Behavior for AI Safety☆18Updated 2 years ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated last year