-
Université de Montréal & Mila
- Montréal, Qc, Canada
- https://orcid.org/0000-0001-8806-7936
Stars
Implemenation of PQMass from Lemos et al. 2024
This is the official code release for Bayesian Flow Networks.
Build scientific simulators, treating them as a directed acyclic graph
PyTorch library for solving imaging inverse problems using deep learning
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
EDM2 and Autoguidance -- Official PyTorch implementation
Learning Diffusion Priors from Observations by Expectation Maximization
PyTorch Implementation of Diffusion Schrodinger Bridge Matching
🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
T-GATE: Temporally Gating Attention to Accelerate Diffusion Model for Free!
High-Resolution Image Synthesis with Latent Diffusion Models
Tiny PyTorch library for maintaining a moving average of a collection of parameters.
Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation
Simple, safe way to store and distribute tensors
FreeU: Free Lunch in Diffusion U-Net (CVPR2024 Oral)
Score-Based Priors for Bayesian Inverse Imaging (ICCV 2023, TMLR 2024)
Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper
A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)
PyTorch implementation of some attentions for Deep Learning Researchers.
You like pytorch? You like micrograd? You love tinygrad! ❤️
Reviews for the Journal of Open Source Software
Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
A gravitational lensing simulator for the machine learning era.
Corner plots, LaTeX tables and plotting walks.
Official Implement of "ADBench: Anomaly Detection Benchmark", NeurIPS 2022.