Uplift modeling and causal inference with machine learning algorithms
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Algorithms for explaining machine learning models
TFDS is a collection of datasets ready to use with TensorFlow,
Jittor is a high-performance deep learning framework
Machine learning, conversational dialog engine for creating chat bots
Open Source Computer Vision Library
Toolkit for conversational AI
Python package for AutoML on Tabular Data with Feature Engineering
Topic Modelling for Humans
Uncover insights, surface problems, monitor, and fine tune your LLM
AI Toolkit for Healthcare Imaging
Deep learning library
Distributed ML Training and Fine-Tuning on Kubernetes
Deep and Machine Learning for Microscopy
Visualizer for neural network, deep learning, machine learning models
Pretrained (Language) Models for Probabilistic Time Series Forecasting
Message Passing Neural Networks for Molecule Property Prediction
Create videos with Stable Diffusion
Hummingbird compiles trained ML models into tensor computation
We write your reusable computer vision tools
JAX-based neural network library
Images to inference with no labeling
Machine learning metrics for distributed, scalable PyTorch application