Multi-class confusion matrix library in Python
A python library for easy manipulation and forecasting of time series
A package for the sparse identification of nonlinear dynamical systems
Geometric deep learning extension library for PyTorch
Classic papers and resources on recommendation
Machine learning software to solve data mining problems
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
Best practices on recommendation systems
DeepMind's software stack for physics-based simulation
A library for accelerating Transformer models on NVIDIA GPUs
Probabilistic reasoning and statistical analysis in TensorFlow
C++-based high-performance parallel environment execution engine
Relax! Flux is the ML library that doesn't make you tensor
A lightweight audio-to-MIDI converter with pitch bend detection
A scientific machine learning (SciML) wrapper for the FEniCS
Natural Gradient Boosting for Probabilistic Prediction
Python library for defining and optimizing mathematical expressions
Medical imaging toolkit for deep learning
State of the Art Natural Language Processing
A Python library for audio
Optax is a gradient processing and optimization library for JAX