Skip to content

@ermongroup ermongroup

Change the repository type filter

All

    Repositories list

    78 repositories

    • cs228-notes

      Public
      Course notes for CS228: Probabilistic Graphical Models.
      SCSS
      •
      MIT License
      •480•2k•6•7•Updated Jun 24, 2025Jun 24, 2025

    • TEOChat

      Public
      Official code for TEOChat, the first vision-language assistant for temporal earth observation data (ICLR 2025).
      Python
      •
      Apache License 2.0
      •6•121•5•0•Updated May 30, 2025May 30, 2025

    • ddim

      Public
      Denoising Diffusion Implicit Models
      Python
      •
      MIT License
      •224•1.7k•14•1•Updated Jul 26, 2024Jul 26, 2024

    • CSDI

      Public
      Codes for "CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputation"
      Jupyter Notebook
      •
      MIT License
      •101•407•7•1•Updated Mar 14, 2024Mar 14, 2024

    • private-rl

      Public
      Code for `Privacy-Constrained Policies via Mutual Information Regularized Policy Gradients', AISTATS 2024
      Python
      •0•0•0•0•Updated Mar 8, 2024Mar 8, 2024

    • ncsn

      Public
      Noise Conditional Score Networks (NeurIPS 2019, Oral)
      generative-modelsneurips-2019score-matchingscore-based-generative-modeling
      Python
      •
      GNU General Public License v3.0
      •115•752•6•3•Updated Feb 14, 2024Feb 14, 2024

    • tile2vec

      Public
      Implementation and examples for Tile2Vec
      Python
      •
      MIT License
      •40•115•4•0•Updated Jan 25, 2024Jan 25, 2024

    • subsets

      Public
      Code for Reparameterizable Subset Sampling via Continuous Relaxations, IJCAI 2019.
      Python
      •8•57•1•2•Updated Oct 3, 2023Oct 3, 2023

    • GraphScoreMatching

      Public
      Official implementation for the paper: Permutation Invariant Graph Generation via Score-Based Generative Modeling
      Python
      •
      GNU General Public License v3.0
      •21•113•3•1•Updated Jul 6, 2023Jul 6, 2023

    • Wifi_Activity_Recognition

      Public
      Code for IEEE Communication Magazine (A Survey on Behaviour Recognition Using WiFi Channle State Information)
      Jupyter Notebook
      •
      GNU General Public License v3.0
      •76•289•19•0•Updated Jun 8, 2023Jun 8, 2023

    • gac

      Public
      Python
      •2•2•0•0•Updated Apr 4, 2023Apr 4, 2023

    • pirank

      Public
      PiRank: Learning to Rank via Differentiable Sorting
      Python
      •
      MIT License
      •9•61•1•2•Updated Mar 24, 2023Mar 24, 2023

    • f-EBM

      Public
      Code for "Training Deep Energy-Based Models with f-Divergence Minimization" ICML 2020
      Python
      •8•36•0•4•Updated Mar 24, 2023Mar 24, 2023

    • neuralsort

      Public
      Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.
      Python
      •
      MIT License
      •26•146•3•3•Updated Mar 24, 2023Mar 24, 2023

    • MetaIRL

      Public
      Meta-Inverse Reinforcement Learning with Probabilistic Context Variables
      Python
      •8•74•3•3•Updated Mar 16, 2023Mar 16, 2023

    • LISA

      Public
      (NeurIPS '22) LISA: Learning Interpretable Skill Abstractions - A framework for unsupervised skill learning using Imitation
      Python
      •6•1•0•0•Updated Feb 23, 2023Feb 23, 2023

    • SDEdit

      Public
      PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations
      pytorchimage-editingimage-manipulationimage-generationscore-matchingcontrollable-generation
      Python
      •
      MIT License
      •94•1.1k•21•2•Updated Feb 12, 2023Feb 12, 2023

    • self-similarity-prior

      Public
      Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations
      Jupyter Notebook
      •
      MIT License
      •4•28•0•0•Updated Nov 26, 2022Nov 26, 2022

    • fairgen

      Public
      Fair Generative Modeling via Weak Supervision
      Jupyter Notebook
      •8•20•1•5•Updated Nov 22, 2022Nov 22, 2022

    • dre-infinity

      Public
      Density Ratio Estimation via Infinitesimal Classification (AISTATS 2022 Oral)
      Python
      •3•20•1•0•Updated Mar 12, 2022Mar 12, 2022

    • BCD-Nets

      Public
      Code for `BCD Nets: Scalable Variational Approaches for Bayesian Causal Discovery`, Neurips 2021
      Python
      •6•26•1•0•Updated Oct 26, 2021Oct 26, 2021

    • HyperSPN

      Public
      PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021
      pytorchregularizationdensity-estimationsum-product-networksprobabilistic-circuits
      Python
      •2•13•0•0•Updated Oct 26, 2021Oct 26, 2021

    • d2c

      Public
      PyTorch implementation of D2C: Diffuison-Decoding Models for Few-shot Conditional Generation.
      Python
      •
      MIT License
      •18•0•0•0•Updated Oct 21, 2021Oct 21, 2021

    • SPN_Variational_Inference

      Public
      PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020
      pytorchgraphical-modelsvariational-inferenceapproximate-inferencesum-product-networksprobabilistic-circuits
      Python
      •2•17•0•0•Updated Oct 11, 2021Oct 11, 2021

    • NDA

      Public
      Python
      •4•24•0•0•Updated Sep 29, 2021Sep 29, 2021

    • fast_feedforward_computation

      Public
      Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
      accelerationpixelcnnrnndensenetnumerical-methodsmadeautoregressive-modelsicml-2021
      Jupyter Notebook
      •1•28•0•0•Updated Sep 25, 2021Sep 25, 2021

    • atlasdemo

      Public
      demo for how to use atlas
      Shell
      •3•1•0•0•Updated Sep 15, 2021Sep 15, 2021

    • PatchDrop

      Public
      PyTorch Implementation of `Learning to Process Fewer Pixels` - [CVPR20 (Oral)]
      Python
      •9•67•2•1•Updated Aug 5, 2021Aug 5, 2021

    • f-dre

      Public
      Featurized Density Ratio Estimation
      Jupyter Notebook
      •5•20•0•0•Updated Jul 11, 2021Jul 11, 2021

    • ncsnv2

      Public
      The official PyTorch implementation for NCSNv2 (NeurIPS 2020)
      generative-modelsdiffusion-modelsneurips-2020score-matchingscore-based-generative-modeling
      Python
      •
      MIT License
      •63•316•9•0•Updated Jun 12, 2021Jun 12, 2021