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 6 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 2 months ago
- ☆31Updated 8 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- GPT implementation in Flax☆18Updated 3 years ago
- Fast and reliable distributed systems in Python☆28Updated 3 months ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆31Updated 2 years ago
- Fast reinforcement learning 💨☆26Updated 3 weeks ago
- PyTorch Package For Quasimetric Learning☆42Updated 9 months ago
- Reinforcement learning in pure JAX.☆13Updated 5 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- Building blocks for productive research☆59Updated last week
- Jax like function transformation engine but micro, microjax☆33Updated 9 months ago
- Causal Analysis of Agent Behavior for AI Safety☆18Updated 2 years ago
- Repo to reproduce the First-Explore paper results☆38Updated 7 months ago
- Clean RL implementation using MLX☆32Updated last year
- General Modules for JAX☆67Updated 4 months ago
- Generalised UDRL☆37Updated 3 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Updated 3 years ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆62Updated 10 months ago
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 5 years ago
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆46Updated 4 years ago
- Accelerated replay buffers in JAX☆43Updated 2 years ago
- Code for the paper "Function-Space Learning Rates"☆23Updated 2 months ago
- ☆23Updated 3 years ago
- 📄Small Batch Size Training for Language Models☆41Updated this week
- ☆33Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 10 months ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆58Updated 3 years ago