Fast and memory-efficient exact attention
Build a machine learning model from a prompt
Spatiotemporal Signal Processing with Neural Machine Learning Models
Topic Modelling for Humans
Decentralized deep learning in PyTorch. Built to train models
Implementation of DeepLabCut
The Triton Inference Server provides an optimized cloud
Python library for portfolio optimization built on top of scikit-learn
A GPU-accelerated library containing highly optimized building blocks
Evaluate and monitor ML models from validation to production
Data manipulation and transformation for audio signal processing
A Python library for learning and evaluating knowledge graph embedding
JAX-based neural network library
Graph Neural Network Library for PyTorch
Environments and algorithms for research in general reinforcement
Collection of various algorithms in mathematics, machine learning
Scalable machine learning for time series forecasting
Low-Rank and Sparse Tools for Background Modeling and Subtraction
Relax! Flux is the ML library that doesn't make you tensor
A scientific machine learning (SciML) wrapper for the FEniCS
Deep learning library
Natural Gradient Boosting for Probabilistic Prediction
A library for accelerating Transformer models on NVIDIA GPUs
Geometric deep learning extension library for PyTorch
A python library for self-supervised learning on images