Message Passing Neural Networks for Molecule Property Prediction
Uplift modeling and causal inference with machine learning algorithms
A python library for self-supervised learning on images
Python package for AutoML on Tabular Data with Feature Engineering
Real-Time High-Resolution Background Matting
Machine learning metrics for distributed, scalable PyTorch application
A Python package for segmenting geospatial data with the SAM
A cross-platform Python library for differentiable programming
An Easy-to-use, Scalable and High-performance RLHF Framework
MiniSom is a minimalistic implementation of the Self Organizing Maps
Reference implementations of MLPerf™ training benchmarks
TimeGPT-1: production ready pre-trained Time Series Foundation Model
Data manipulation and transformation for audio signal processing
An easy-to-use & supercharged open-source experiment tracker
SkyPilot: Run AI and batch jobs on any infra
Investment Research for Everyone, Everywhere
Machine Learning automation and tracking
Core ML tools contain supporting tools for Core ML model conversion
ONNX-TensorRT: TensorRT backend for ONNX
Streamline your ML workflow
Open deep learning compiler stack for cpu, gpu, etc.
Superfast AI decision making and processing of multi-modal data
A system for quickly generating training data with weak supervision
Machine Learning Pipelines for Kubeflow
An MLOps framework to package, deploy, monitor and manage models