Test Suites for validating ML models & data
Easy-to-use,Modular and Extendible package of deep-learning models
JAX-based neural network library
Build portable, production-ready MLOps pipelines
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
A GPU-accelerated library containing highly optimized building blocks
mlpack: a scalable C++ machine learning library
Making large AI models cheaper, faster and more accessible
The easiest way to use deep metric learning in your application
Graph Neural Network Library for PyTorch
Package of deep-learning based CTR models
Open deep learning compiler stack for cpu, gpu, etc.
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
AI agents autonomously run and improve ML experiments overnight
Python examples of popular machine learning algorithms
A scientific machine learning (SciML) wrapper for the FEniCS
An MLOps framework to package, deploy, monitor and manage models
Nx-powered Neural Networks
Proofs, cases, concept supplements, and reference explanations
Detecting silent model failure. NannyML estimates performance
A library for scientific machine learning & physics-informed learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Training data (data labeling, annotation, workflow) for all data types
Petastorm library enables single machine or distributed training