A fast library for AutoML and tuning
Simulation of spiking neural networks (SNNs) using PyTorch
Differentiable SDE solvers with GPU support and efficient sensitivity
A library for scientific machine learning & physics-informed learning
A scikit-learn compatible neural network library that wraps PyTorch
A cross-platform Python library for differentiable programming
Neural machine translation and sequence learning using TensorFlow
JAX-based neural network library
Embed images and sentences into fixed-length vectors
Deep learning library
Deep and Machine Learning for Microscopy
Fast image augmentation library and an easy-to-use wrapper
Graph Neural Network Library for PyTorch
Python library for Representation Learning on Knowledge Graphs
Spatiotemporal Signal Processing with Neural Machine Learning Models
Training PyTorch models with differential privacy
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
A python library for easy manipulation and forecasting of time series
A Python package for extending the official PyTorch
Python toolbox to create adversarial examples
JAX-based neural network library
Build cross-modal and multimodal applications on the cloud
Library to help with training and evaluating neural networks
Optax is a gradient processing and optimization library for JAX
Geometric deep learning extension library for PyTorch