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
Flash enables you to easily configure and run complex AI recipes
OpenMMLab Text Detection, Recognition and Understanding Toolbox
A Python library that helps data scientists to infer causation
Distributed training framework for TensorFlow, Keras, PyTorch, etc.
Learning to rank in TensorFlow
Graph Neural Networks with Keras and Tensorflow 2
Official Python Implementation for "3D Multi-Object Tracking
TensorFlow documentation
Benchmark datasets, data loaders, and evaluators for graph machine
A library for graph deep learning research
Fast Forward Computer Vision (and other ML workloads!)
Sequence-to-sequence framework, focused on Neural Machine Translation
A Machine Learning Framework for Time Series Intelligence
Mars is a tensor-based unified framework for large-scale data
Build and deploy machine learning microservices
An API Oriented Open-source Python Framework for Unsupervised Learning
Human Activity Recognition example using TensorFlow on smartphone
Large-scale pretraining for dialogue