yilundu / irem_code_release
ICML 2022: Learning Iterative Reasoning through Energy Minimization
☆46Updated last year
Alternatives and similar repositories for irem_code_release:
Users that are interested in irem_code_release are comparing it to the libraries listed below
- NF-Layers for constructing neural functionals.☆82Updated last year
- ☆51Updated 8 months ago
- PyTorch Package For Quasimetric Learning☆41Updated 3 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆48Updated 2 months ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆17Updated last year
- ☆28Updated 2 months ago
- ☆30Updated 2 months ago
- Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"☆39Updated 7 months ago
- ☆56Updated 2 years ago
- ☆26Updated last year
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction (ICML 2023)☆46Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated last year
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆48Updated last year
- ☆23Updated 6 months ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆81Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆82Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆53Updated 3 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆24Updated 8 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆95Updated last year
- ☆35Updated last year
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆87Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆30Updated 6 months ago
- ☆15Updated 2 years ago