Hummingbird compiles trained ML models into tensor computation
Embed images and sentences into fixed-length vectors
Open standard for machine learning interoperability
Stanford NLP Python library for many human languages
JAX-based neural network library
Core ML tools contain supporting tools for Core ML model conversion
A general-purpose probabilistic programming system
oneAPI Deep Neural Network Library (oneDNN)
Neural Network Compression Framework for enhanced OpenVINO
Message Passing Neural Networks for Molecule Property Prediction
Implementation of Imagen, Google's Text-to-Image Neural Network
A library for scientific machine learning & physics-informed learning
Graph Neural Network Library for PyTorch
Internet-scale Neural Networks
Implementation of 'lightweight' GAN, proposed in ICLR 2021
DeepVariant is an analysis pipeline that uses a deep neural networks
Face recognition with deep neural networks
OpenMMLab Model Deployment Framework
Differentiable SDE solvers with GPU support and efficient sensitivity
A cross-platform Python library for differentiable programming
Build high-performance AI models with modular building blocks
Neural Search
C++ Implementation of PyTorch Tutorials for Everyone
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Training PyTorch models with differential privacy