#deep-learning

Probabilistic time series modeling in Python

DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning

Fast and accurate AI powered file content types detection

Pytorch domain library for recommendation systems

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning. Ongoing development and maintenance by t2k.

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

A game theoretic approach to explain the output of any machine learning model.

Supervisely SDK for Python - convenient way to automate, customize and extend Supervisely Platform for your computer vision task

A unified library for building, evaluating, and storing speculative decoding algorithms for LLM inference in vLLM

Speed up model training by fixing data loading.

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.

⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

Deep learning software to decode EEG, ECG or MEG signals

ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution

Python suite for neuroscience research across all modalities.

Stop trusting your pipeline. Verify it; visually on a canvas or in Python. Self-hosted platform + library.

Framework agnostic sliced/tiled inference + interactive ui + error analysis plots

GeoAI: Artificial Intelligence for Geospatial Data

Image annotation with Python. Supports polygon, rectangle, circle, line, point, and AI-assisted annotation.