tinkoff-ai / probabilistic-embeddingsLinks
"Probabilistic Embeddings Revisited" paper official repository
☆28Updated 2 years ago
Alternatives and similar repositories for probabilistic-embeddings
Users that are interested in probabilistic-embeddings are comparing it to the libraries listed below
Sorting:
- Repository for "Generative Flow Networks as Entropy-Regularized RL" (AISTATS-2024, Oral)☆35Updated last year
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆75Updated 2 years ago
- Official implementation for "Q-Ensemble for Offline RL: Don't Scale the Ensemble, Scale the Batch Size", NeurIPS 2022, Offline RL Worksho…☆21Updated 2 years ago
- Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"☆31Updated 11 months ago
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆55Updated 2 years ago
- Code for the paper "Showing Your Offline Reinforcement Learning Work: Online Evaluation Budget Matters", ICML 2022☆28Updated 3 years ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆40Updated 2 years ago
- Official code for "Can Wikipedia Help Offline Reinforcement Learning?" by Machel Reid, Yutaro Yamada and Shixiang Shane Gu☆105Updated 3 years ago
- The original PyTorch implementation of the "EXACT: How Train Your Accuracy"☆10Updated 2 years ago
- ☆37Updated 2 years ago
- Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆18Updated last year
- ☆56Updated 2 years ago
- Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023☆53Updated 2 years ago
- Official implementation of Transformer Neural Processes☆78Updated 3 years ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆78Updated 6 months ago
- This repository contains the code for our paper "Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguo…☆41Updated 2 years ago
- Clean, extensible implementation of MACAW [ICML 2021]☆12Updated 3 years ago
- A reinforcement learning environment for the IGLU 2022 at NeurIPS☆34Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆52Updated 2 years ago
- Generalised UDRL☆37Updated 3 years ago
- ☆31Updated 4 years ago
- Code for the paper "PALBERT: Teaching ALBERT to Ponder", NeurIPS 2022 Spotlight☆37Updated 2 years ago
- Official implementation for "Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flows", NeurIPS 2022, O…☆12Updated 2 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Updated 2 years ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Updated 2 years ago
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago
- GULAG: GUessing LAnGuages with neural networks☆13Updated 3 years ago
- Implementations of Temporal Difference InfoNCE (TD InfoNCE)☆30Updated last year
- ☆52Updated 4 years ago
- ☆22Updated 5 months ago