A library for scientific machine learning & physics-informed learning
Training data (data labeling, annotation, workflow) for all data types
Python package built to ease deep learning on graph
Toolkit for conversational AI
Hummingbird compiles trained ML models into tensor computation
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Open source machine learning framework to automate text conversations
Python Optimal Transport
Online machine learning in Python
Training and deploying machine learning models on Amazon SageMaker
Tensor Learning in Python
Natural Gradient Boosting for Probabilistic Prediction
A Python library for learning and evaluating knowledge graph embedding
Elyra extends JupyterLab with an AI centric approach
Topic Modelling for Humans
Pretrained (Language) Models for Probabilistic Time Series Forecasting
A high performance implementation of HDBSCAN clustering
Simulation of spiking neural networks (SNNs) using PyTorch
Practice implementing softmax, attention, GPT-2 and more
Foundation Model for Tabular Data
Machine learning metrics for distributed, scalable PyTorch application
Clean, Robust, and Unified PyTorch implementation
Uplift modeling and causal inference with machine learning algorithms
DoWhy is a Python library for causal inference
Numerical differential equation solvers in JAX