FAIR Chemistry's library of machine learning methods for chemistry
JAX-based neural network library
A cross-platform Python library for differentiable programming
A Graph Neural Network Library in Jax
A model library for exploring state-of-the-art deep learning
mixup: Beyond Empirical Risk Minimization
Pretty Tensor: Fluent Networks in TensorFlow