danijar / portalLinks
Fast and reliable distributed systems in Python
☆28Updated 3 months ago
Alternatives and similar repositories for portal
Users that are interested in portal are comparing it to the libraries listed below
Sorting:
- Building blocks for productive research☆59Updated last week
- PyTorch Package For Quasimetric Learning☆42Updated 9 months ago
- Tools and Utils for Experiments (TUX)☆16Updated 6 months ago
- Fast dataset format and loader☆22Updated 6 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- General Modules for JAX☆67Updated 4 months ago
- Galactic Scaling End-to-End Reinforcement Learning for Rearrangement at 100k Steps-Per-Second☆84Updated 2 years ago
- Fast reinforcement learning research☆61Updated 8 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Updated 2 years ago
- AgentHive provides the primitives and helpers for a seamless usage of robohive within TorchRL.☆34Updated last year
- Configuration with Dataclasses+YAML+Argparse. Fork of Pyrallis☆58Updated 3 months ago
- A simple, performant and scalable JAX-based world modeling codebase☆58Updated this week
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆73Updated last year
- flexible meta-learning in jax☆14Updated last year
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆32Updated last year
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated last year
- Accelerated replay buffers in JAX☆43Updated 2 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆83Updated last year
- Bingham Policy Parameterization for 3D Rotations in Reinforcement Learning☆24Updated 3 years ago
- ☆115Updated 2 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆114Updated 11 months ago
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆68Updated 11 months ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆70Updated last year
- ☆19Updated 2 years ago
- Code for the paper Task Agnostic Morphology Evolution.☆20Updated 4 years ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated 2 years ago
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆20Updated last year
- Open source code for paper "Denoised MDPs: Learning World Models Better Than the World Itself"☆134Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆58Updated 3 years ago
- An implementation of MuZero in JAX.☆56Updated 2 years ago