Geometric deep learning extension library for PyTorch
Classic papers and resources on recommendation
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
DeepMind's software stack for physics-based simulation
Probabilistic reasoning and statistical analysis in TensorFlow
A lightweight audio-to-MIDI converter with pitch bend detection
Relax! Flux is the ML library that doesn't make you tensor
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
Optax is a gradient processing and optimization library for JAX
A Python library for audio
Fast Python collaborative filtering for implicit feedback datasets
A Python library for audio data augmentation
The Operator Splitting QP Solver
A Hyperparameter Tuning Library for Keras
Python binding to the Apache Tika™ REST services
High-Performance Face Recognition Library on PaddlePaddle & PyTorch
Advanced AI Explainability for computer vision
Nx-powered Neural Networks