Python package built to ease deep learning on graph
Graph Neural Network Library for PyTorch
A unified interface for distributed computing
Pre-built implicit layer architectures with O(1) backprop, GPUs
Elyra extends JupyterLab with an AI centric approach
Python package for AutoML on Tabular Data with Feature Engineering
RAPIDS Machine Learning Library
TFX is an end-to-end platform for deploying production ML pipelines
A library for audio and music analysis, feature extraction
A high performance implementation of HDBSCAN clustering
Scalable machine learning for time series forecasting
A python library for easy manipulation and forecasting of time series
Powering Amazon custom machine learning chips
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
JAX-based neural network library
Uplift modeling and causal inference with machine learning algorithms
A Flexible and Powerful Parameter Server for large-scale ML
Machine Learning Toolkit for Kubernetes
OpenMLDB is an open-source machine learning database
A system for quickly generating training data with weak supervision
A Python Automated Machine Learning tool that optimizes ML
Deep and Machine Learning for Microscopy
Package that makes it trivial to create and evaluate machine learning
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Trainable models and NN optimization tools