A simple and efficient Mamba implementation in pure PyTorch and MLX.

Structured state space sequence models

Mamba SSM architecture

An extremely fast Python package and project manager, written in Rust.

MoDL: Model-Based Deep Learning Architecture for Inverse Problems

[NeurIPS'19] Deep Equilibrium Models

A multi-contrast multi-repetition multi-channel MRI k-space dataset for low-field MRI research

Code for random Fourier features based on Rahimi and Recht's 2007 paper.

A didactic Python library with well-commented and annotated implementations of machine learning algorithms.

DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning

A curated list of awesome computer vision resources

A curated list of resources on implicit neural representations.

Python package for signal processing, with emphasis on iterative methods

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

Text-to-3D & Image-to-3D & Mesh Exportation with NeRF + Diffusion.

Implementation of "SVDiff: Compact Parameter Space for Diffusion Fine-Tuning"

AI Toolkit for Healthcare Imaging

Using Low-rank adaptation to quickly fine-tune diffusion models.

High-Resolution Image Synthesis with Latent Diffusion Models

LLM inference in C/C++

A latent text-to-image diffusion model

A PyTorch-powered differentiable image reconstruction/optimization toolbox

Efficiently computes derivatives of NumPy code.

Code repo for the paper BiT Robustly Binarized Multi-distilled Transformer

A book teaching assembly language programming on the ARM 64 bit ISA. Along the way, good programming practices and insights into code development are offered which apply directly to higher level la…

Code and experiments for the MIMOSA Stealthy Malware Analysis System

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!