Stars
A toolbox to iNNvestigate neural networks' predictions!
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Open source platform for the machine learning lifecycle
Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations
NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference i…
Algorithms for outlier, adversarial and drift detection
FinRL: Financial Reinforcement Learning. 🔥
Barbershop: GAN-based Image Compositing using Segmentation Masks (SIGGRAPH Asia 2021)
A comprehensive machine learning repository containing 30+ notebooks on different concepts, algorithms and techniques.
🎓 Sharing machine learning course / lecture notes.
Easy-to-use Speech Toolkit including Self-Supervised Learning model, SOTA/Streaming ASR with punctuation, Streaming TTS with text frontend, Speaker Verification System, End-to-End Speech Translatio…
For slim progress bars like on YouTube, Medium, etc
Scrapy, a fast high-level web crawling & scraping framework for Python.
It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced research…
A collection of Jupyter notebooks to play with NVIDIA's StyleGAN3 and OpenAI's CLIP for a text-based guided image generation.
💯 Curated coding interview preparation materials for busy software engineers
Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and mo…
Explain, analyze, and visualize NLP language models. Ecco creates interactive visualizations directly in Jupyter notebooks explaining the behavior of Transformer-based language models (like GPT2, B…
Convert Machine Learning Code Between Frameworks
Distributed and decentralized training framework for PyTorch over graph
GLIDE: a diffusion-based text-conditional image synthesis model
🧠 A study guide to learn about Transformers
🟠 A study guide to learn about Graph Neural Networks (GNNs)
A course in reinforcement learning in the wild
Framework agnostic sliced/tiled inference + interactive ui + error analysis plots