A library for scientific machine learning & physics-informed learning
Implementation of Imagen, Google's Text-to-Image Neural Network
Graph Neural Network Library for PyTorch
Geometric deep learning extension library for PyTorch
A cross-platform Python library for differentiable programming
Message Passing Neural Networks for Molecule Property Prediction
Python package built to ease deep learning on graph
Training PyTorch models with differential privacy
Package of deep-learning based CTR models
Neural Search
A scikit-learn compatible neural network library that wraps PyTorch
Audio generation using diffusion models, in PyTorch
Neural machine translation and sequence learning using TensorFlow
MTEB: Massive Text Embedding Benchmark
The data structure for multimodal data
ManimML is a project focused on providing animations
Task-oriented finetuning for better embeddings on neural search
Fast image augmentation library and an easy-to-use wrapper
Stanford NLP Python library for many human languages
A simple forecasting package
Open Source Differentiable Computer Vision Library
Hummingbird compiles trained ML models into tensor computation
A fast embedded library for approximate nearest neighbor search
State-of-the-art Multilingual Question Answering research
Deep learning framework