yilundu / irem_code_releaseLinks
ICML 2022: Learning Iterative Reasoning through Energy Minimization
☆46Updated 2 years ago
Alternatives and similar repositories for irem_code_release
Users that are interested in irem_code_release are comparing it to the libraries listed below
Sorting:
- NF-Layers for constructing neural functionals.☆85Updated last year
- ☆65Updated 11 months ago
- PyTorch Package For Quasimetric Learning☆42Updated 7 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆29Updated 11 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 6 months ago
- Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"☆42Updated 10 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- ☆56Updated 2 years ago
- Building blocks for productive research☆55Updated 4 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆45Updated 2 years ago
- ☆29Updated 6 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- ☆28Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆63Updated 3 years ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆26Updated this week
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 2 years ago
- ☆51Updated 4 years ago
- ☆51Updated last year
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”☆76Updated last year
- Code release for "Stochastic Optimal Control Matching"☆35Updated 9 months ago
- Implementation of PSGD optimizer in JAX☆33Updated 5 months ago
- ☆31Updated 7 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆101Updated last year
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆60Updated 3 months ago
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction☆30Updated last year
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆61Updated 2 years ago
- Baselines for Model-Based Optimization☆53Updated 3 years ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆57Updated last year
- [ICLR 2022] Path integral sampler☆46Updated last year