Embed images and sentences into fixed-length vectors
Fast ODE Solver for Diffusion Probabilistic Model Sampling
Serve machine learning models within a Docker container
The PyTorch-based audio source separation toolkit for researchers
Backtest and optimize your ML trading strategies with only 3 lines
Differentiable SDE solvers with GPU support and efficient sensitivity
PyTorch Lightning + Hydra. A very user-friendly template
Quantitative analysis, strategies and backtests
User-friendly library to find similar objects
Meta-Transformer for Unified Multimodal Learning
Visual localization made easy with hloc
Interpretability and explainability of data and machine learning model
A Python library that helps data scientists to infer causation
Flash enables you to easily configure and run complex AI recipes
OpenMMLab Text Detection, Recognition and Understanding Toolbox
Distributed training framework for TensorFlow, Keras, PyTorch, etc.
Learning to rank in TensorFlow
TensorFlow documentation
Graph Neural Networks with Keras and Tensorflow 2
Official Python Implementation for "3D Multi-Object Tracking
Benchmark datasets, data loaders, and evaluators for graph machine
A library for graph deep learning research
Fast Forward Computer Vision (and other ML workloads!)
A Machine Learning Framework for Time Series Intelligence
Sequence-to-sequence framework, focused on Neural Machine Translation