Deep learning library
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
A Python package for segmenting geospatial data with the SAM
Topic Modelling for Humans
PyTorch extensions for fast R&D prototyping and Kaggle farming
Visualizer for neural network, deep learning, machine learning models
TFX is an end-to-end platform for deploying production ML pipelines
Open source machine learning framework to automate text conversations
Message Passing Neural Networks for Molecule Property Prediction
Explainability and Interpretability to Develop Reliable ML models
Hummingbird compiles trained ML models into tensor computation
We write your reusable computer vision tools
Flower: A Friendly Federated Learning Framework
Open source platform for the machine learning lifecycle
An extensive node suite that enables ComfyUI to process 3D inputs
Learn how to develop, deploy and iterate on production-grade ML
Medical imaging toolkit for deep learning
Fast image augmentation library and an easy-to-use wrapper
Machine learning metrics for distributed, scalable PyTorch application
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
Implementation of DeepLabCut
A Pythonic framework to simplify AI service building
A unified framework for machine learning with time series
Functional Machine Learning