sobieskibj

publications_2024

IDEAS scientific achievements

eXplainableMachineLearning-2025

eXplainable Machine Learning 2024/25 at MIM UW

DiG-IN

DiG-IN: Diffusion Guidance for Investigating Networks - Uncovering Classifier Differences, Neuron Visualisations, and Visual Counterfactual Explanations

infd

Image Neural Field Diffusion Models, CVPR 2024 (Highlight)

grand-challenge.org

A platform for end-to-end development of machine learning solutions in biomedical imaging

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

FastDiME_Med

ect

Consistency Models Made Easy

latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models

deep-tempest

Restoration for TEMPEST images using deep-learning

score_sde

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

micrograd

The Autograd Engine

rectified-flow-pytorch

Implementation of rectified flow and some of its followup research / improvements in Pytorch

MetaQuantus

MetaQuantus is an XAI performance tool to identify reliable evaluation metrics

I2SB

diti

Code release for ICLR 2024 paper: Exploring Diffusion Time-steps for Unsupervised Representation Learning

depyf

depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.

cs228-notes

Course notes for CS228: Probabilistic Graphical Models.

unet.cu

UNet diffusion model in pure CUDA

score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Soft-Truncation

A repo for "Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation"

rrclarc

Right-Reasons Class Artifact Compensation

MCG_diffusion

Official PyTorch implementation of the NeurIPS 2022 paper "Improving Diffusion Models for Inverse Problems using Manifold Constraints (https://arxiv.org/abs/2206.00941)"

saliency

Framework-agnostic implementation for state-of-the-art saliency methods (XRAI, BlurIG, SmoothGrad, and more).

ema-pytorch

A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model

Awesome-Consistency-Models

Awesome List of Consistency Models

Awesome-explainable-AI

A collection of research materials on explainable AI/ML

shapiq

Shapley Interactions for Machine Learning

wandb

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

RectifiedFlow

Official Implementation of Rectified Flow (ICLR2023 Spotlight)