Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
AI discovers 520000 stable inorganic crystal structures for research
Nx-powered Neural Networks
A library for scientific machine learning & physics-informed learning
A general-purpose probabilistic programming system
AI Toolkit for Healthcare Imaging
Tool for visualizing and tracking your machine learning experiments
Resources for deep learning with satellite & aerial imagery
A PyTorch-based Speech Toolkit
Python binding to the Apache Tika™ REST services
Weaviate is a cloud-native, modular, real-time vector search engine
Simple and distributed Machine Learning
Evaluate and monitor ML models from validation to production
C++ image processing and machine learning library with using of SIMD
A retargetable MLIR-based machine learning compiler runtime toolkit
A library for easily evaluating machine learning models and datasets
A scikit-learn compatible neural network library that wraps PyTorch
Traditional machine learning on top of Nx
Message Passing Neural Networks for Molecule Property Prediction
Hummingbird compiles trained ML models into tensor computation
Machine learning metrics for distributed, scalable PyTorch application
Functional Machine Learning
Bayesian inference with probabilistic programming