A Python vector database you just need, no more, no less
Differentiable SDE solvers with GPU support and efficient sensitivity
Decentralized deep learning in PyTorch. Built to train models
MTEB: Massive Text Embedding Benchmark
Easy-to-use and powerful NLP library with Awesome model zoo
Segmentation models with pretrained backbones. PyTorch
The data structure for multimodal data
Build cross-modal and multimodal applications on the cloud
Build AI-powered semantic search applications
Library to help with training and evaluating neural networks
Advanced evolutionary computation library built on top of PyTorch
Build high-performance AI models with modular building blocks
MiniSom is a minimalistic implementation of the Self Organizing Maps
Optax is a gradient processing and optimization library for JAX
Deep learning library
Automated Machine Learning on Kubernetes
User toolkit for analyzing and interfacing with Large Language Models
A fast library for AutoML and tuning
Deep and Machine Learning for Microscopy
Framework that is dedicated to making neural data processing
Geometric deep learning extension library for PyTorch
Multi-Modal Neural Networks for Semantic Search, based on Mid-Fusion
The Model Zoo of cognitive diagnosis models
Deep neural networks for density functional theory Hamiltonian
Visual localization made easy with hloc