Stars
Refine high-quality datasets and visual AI models
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 …
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
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"
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
Probabilistic reasoning and statistical analysis in TensorFlow
Tensors and Dynamic neural networks in Python with strong GPU acceleration