Simple and distributed Machine Learning
Visualizer for neural network, deep learning, machine learning models
A cross-platform Python library for differentiable programming
Predicting stock prices using a TensorFlow LSTM
oneAPI Deep Neural Network Library (oneDNN)
C++ library for high performance inference on NVIDIA GPUs
A library for scientific machine learning & physics-informed learning
TensorFlow.js is a library for machine learning in JavaScript
Graph Neural Network Library for PyTorch
A generic, simple and fast implementation of Deepmind's AlphaZero
Nx-powered Neural Networks
High-performance neural network inference framework for mobile
JAX-based neural network library
AIMET is a library that provides advanced quantization and compression
Training PyTorch models with differential privacy
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Lightning fast C++/CUDA neural network framework
A general-purpose probabilistic programming system
FAIR Chemistry's library of machine learning methods for chemistry
Implementation of Imagen, Google's Text-to-Image Neural Network
Set of comprehensive computer vision & machine intelligence libraries
Pre-trained Neural Network models in Axon
High-efficiency floating-point neural network inference operators
A JAX research toolkit to build, edit, & visualize neural networks
OpenMMLab Model Deployment Framework