Refine high-quality datasets and visual AI models

GENIE: Higher-Order Denoising Diffusion Solvers

18.336 - Fast Methods for Partial Differential and Integral Equations

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

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Score-Based Generative Modeling with Critically-Damped Langevin Diffusion

[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.

You like pytorch? You like micrograd? You love tinygrad! ❤️

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

Matplotlib styles for scientific plotting

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

sbi is a Python package for simulation-based inference, designed to meet the needs of both researchers and practitioners. Whether you need fine-grained control or an easy-to-use interface, sbi has …

Sequential Neural Likelihood

Sum Product Flow: An Easy and Extensible Library for Sum-Product Networks

t-sne visualization of mnist images when feature is represented by raw pixels and cnn learned feature

A python wrapper for Barnes-Hut tsne

Real NVP PyTorch a Minimal Working Example | Normalizing Flow

Bayesian Deep Learning Benchmarks

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".

Code for "Latent ODEs for Irregularly-Sampled Time Series" paper

Code for our NeurIPS 2019 *spotlight* "Provably Robust Deep Learning via Adversarially Trained Smoothed Classifiers"

Gaussian processes in TensorFlow

TikZ library for drawing Bayesian networks, graphical models and (directed) factor graphs in LaTeX.

Pytorch implementation of Augmented Neural ODEs 🌻

Tensorflow implementation of a Neural Turing Machine

SciPy library main repository

Probabilistic reasoning and statistical analysis in TensorFlow

Tensors and Dynamic neural networks in Python with strong GPU acceleration