Open deep learning compiler stack for cpu, gpu, etc.
Powering Amazon custom machine learning chips
Machine learning metrics for distributed, scalable PyTorch application
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, etc
Pretrained (Language) Models for Probabilistic Time Series Forecasting
Toolkit for conversational AI
Flower: A Friendly Federated Learning Framework
An open source implementation of CLIP
The most intuitive, flexible, way for researchers to build models
Train machine learning models within Docker containers
A Python library for learning and evaluating knowledge graph embedding
The machine learning toolkit for time series analysis in Python
A library for easily evaluating machine learning models and datasets
A Python package to assess and improve fairness of ML models
Hummingbird compiles trained ML models into tensor computation
Python Package for ML-Based Heterogeneous Treatment Effects Estimation
A Python library for audio
A Pythonic framework to simplify AI service building
We write your reusable computer vision tools
This repos contains notebooks for the Advanced Solutions Lab
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Qlib is an AI-oriented quantitative investment platform
Deep and Machine Learning for Microscopy
Trainable models and NN optimization tools
Implementation of DeepLabCut