Multi-class confusion matrix library in Python
C++ library for high performance inference on NVIDIA GPUs
Implementation of DeepLabCut
A python library for easy manipulation and forecasting of time series
A package for the sparse identification of nonlinear dynamical systems
Classic papers and resources on recommendation
A lightweight audio-to-MIDI converter with pitch bend detection
Geometric deep learning extension library for PyTorch
Scalable machine learning for time series forecasting
A Python library for learning and evaluating knowledge graph embedding
A python library for self-supervised learning on images
Python package built to ease deep learning on graph
A library for accelerating Transformer models on NVIDIA GPUs
Probabilistic reasoning and statistical analysis in TensorFlow
Python library for defining and optimizing mathematical expressions
A scientific machine learning (SciML) wrapper for the FEniCS
Optax is a gradient processing and optimization library for JAX
Library to help with training and evaluating neural networks
Relax! Flux is the ML library that doesn't make you tensor
DeepMind's software stack for physics-based simulation
Natural Gradient Boosting for Probabilistic Prediction
A Python library for audio
Medical imaging toolkit for deep learning
Python binding to the Apache Tika™ REST services