Stars
Multi-LoRA inference server that scales to 1000s of fine-tuned LLMs
Perceptual video quality assessment based on multi-method fusion.
An MLOps framework to package, deploy, monitor and manage thousands of production machine learning models
Rembg is a tool to remove images background
The image prompt adapter is designed to enable a pretrained text-to-image diffusion model to generate images with image prompt.
LAVIS - A One-stop Library for Language-Vision Intelligence
Lora beYond Conventional methods, Other Rank adaptation Implementations for Stable diffusion.
PyTorch code to run synthetic experiments.
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
PyTorch package for the discrete VAE used for DALL·E.
The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
Optimization with orthogonal constraints and on general manifolds
Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
Mellotron: a multispeaker voice synthesis model based on Tacotron 2 GST that can make a voice emote and sing without emotive or singing training data
code for "Residual Flows for Invertible Generative Modeling".
Make huge neural nets fit in memory
code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".
Probabilistic reasoning and statistical analysis in TensorFlow
A probabilistic programming language in TensorFlow. Deep generative models, variational inference.
A Flow-based Generative Network for Speech Synthesis
Progressive Growing of GANs for Improved Quality, Stability, and Variation