danijar / portal
Fast and reliable distributed systems in Python
☆25Updated 2 months ago
Alternatives and similar repositories for portal:
Users that are interested in portal are comparing it to the libraries listed below
- PyTorch Package For Quasimetric Learning☆41Updated 5 months ago
- Building blocks for productive research☆52Updated 2 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- General Modules for JAX☆64Updated 2 weeks ago
- Tools and Utils for Experiments (TUX)☆16Updated 2 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆28Updated 10 months ago
- Bingham Policy Parameterization for 3D Rotations in Reinforcement Learning☆24Updated 3 years ago
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆20Updated 9 months ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆73Updated last year
- Fast dataset format and loader☆22Updated 3 months ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16Updated last year
- Fast reinforcement learning research☆59Updated 4 months ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆14Updated 3 weeks ago
- flexible meta-learning in jax☆12Updated last year
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆27Updated 5 months ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆53Updated last year
- ☆31Updated last year
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Code for Continual Learning of Control Primitives☆18Updated 4 years ago
- Open source code for paper "Denoised MDPs: Learning World Models Better Than the World Itself"☆136Updated last year
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆21Updated 11 months ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- ☆81Updated 2 years ago
- ☆42Updated 2 years ago
- Experiment. Plot. Tabulate.☆69Updated 7 months ago
- Standalone library of frequently-used wrappers for dm_env environments.☆17Updated 9 months ago