A unified framework for machine learning with time series
A Lightweight Face Recognition and Facial Attribute Analysis
MiniSom is a minimalistic implementation of the Self Organizing Maps
Optax is a gradient processing and optimization library for JAX
Fault-tolerant, highly scalable GPU orchestration
SkyPilot: Run AI and batch jobs on any infra
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Trainable models and NN optimization tools
Streamline your ML workflow
Flash enables you to easily configure and run complex AI recipes
The open-source tool for building high-quality datasets
Train machine learning models within Docker containers
Single-cell analysis in Python
Jittor is a high-performance deep learning framework
Gaussian processes in TensorFlow
Open deep learning compiler stack for cpu, gpu, etc.
TFDS is a collection of datasets ready to use with TensorFlow,
Message Passing Neural Networks for Molecule Property Prediction
A Python package for extending the official PyTorch
A general-purpose probabilistic programming system
A Pythonic framework to simplify AI service building
Topic Modelling for Humans
We write your reusable computer vision tools
Pytorch domain library for recommendation systems
A fast library for AutoML and tuning