Python framework for adversarial attacks, and data augmentation
Python library for decision tree visualization & model interpretation
Interactively analyze ML models to understand their behavior
SkyPilot: Run AI and batch jobs on any infra
We write your reusable computer vision tools
MLOps simplified. From ML Pipeline ⇨ Data Product without the hassle
Pytorch domain library for recommendation systems
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Simulation of spiking neural networks (SNNs) using PyTorch
Data manipulation and transformation for audio signal processing
A Pythonic framework to simplify AI service building
Adversarial Robustness Toolbox (ART) - Python Library for ML security
A unified framework for machine learning with time series
Lightning fast C++/CUDA neural network framework
A python library for self-supervised learning on images
DeepMind's software stack for physics-based simulation
Practice implementing softmax, attention, GPT-2 and more
This repository is for helping those interested in machine learning
Low-Rank and Sparse Tools for Background Modeling and Subtraction
A Python package for extending the official PyTorch
An unsupervised and free tool for image and video dataset analysis
Physical Symbolic Optimization
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
The Compute Library is a set of computer vision and machine learning
Machine learning metrics for distributed, scalable PyTorch application