Implementation of 'lightweight' GAN, proposed in ICLR 2021
A scikit-learn compatible neural network library that wraps PyTorch
Hummingbird compiles trained ML models into tensor computation
Machine learning metrics for distributed, scalable PyTorch application
A Python Package to Tackle the Curse of Imbalanced Datasets in ML
Training and deploying machine learning models on Amazon SageMaker
Statistical library designed to fill the void in Python's time series
Gaussian processes in TensorFlow
This repos contains notebooks for the Advanced Solutions Lab
A Hyperparameter Tuning Library for Keras
Multi-class confusion matrix library in Python
Solve end to end problems using Llama model family
A TensorFlow library for applied reinforcement learning
TFX is an end-to-end platform for deploying production ML pipelines
Pretrained (Language) Models for Probabilistic Time Series Forecasting
SkyPilot: Run AI and batch jobs on any infra
MLOps simplified. From ML Pipeline ⇨ Data Product without the hassle
Pytorch domain library for recommendation systems
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Data manipulation and transformation for audio signal processing
Adversarial Robustness Toolbox (ART) - Python Library for ML security
A python library for self-supervised learning on images
Practice implementing softmax, attention, GPT-2 and more
A Python package for extending the official PyTorch
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method