Stars
🦜🔗 Build context-aware reasoning applications
An open source implementation of CLIP.
Geometric loss functions between point clouds, images and volumes
Algorithms for outlier, adversarial and drift detection
KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
Goven (go-oven) is a go library that allows you to have a drop-in query language for your database schema.
A powerful and flexible machine learning platform for drug discovery
scikit-learn: machine learning in Python
SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]
Model interpretability and understanding for PyTorch
Therapeutics Commons (TDC-2): Multimodal Foundation for Therapeutic Science
An inference server for your machine learning models, including support for multiple frameworks, multi-model serving and more
Graph Neural Network Library for PyTorch
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
PyTorch implementations of Generative Adversarial Networks.
Corruption and Perturbation Robustness (ICLR 2019)
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX