Deep universal probabilistic programming with Python and PyTorch
JAX-based neural network library
A cross-platform Python library for differentiable programming
TensorFlow-based neural network library
Python library for model interpretation/explanations
Neural network library for developers