#deep-learning

The official Roboflow Python package. Manage your datasets, models, and deployments. Roboflow has everything you need to build a computer vision application.

A python library for user-friendly forecasting and anomaly detection on time series.

ZenML 🙏: One AI Platform from Pipelines to Agents. https://zenml.io.

The easiest way to serve AI apps and models - Build Model Inference APIs, Job queues, LLM apps, Multi-model pipelines, and more!

Structured Outputs

Pretrained model hub for Keras 3.

A python library for self-supervised learning on images.

Low-code framework for building custom LLMs, neural networks, and other AI models

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

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

We write your reusable computer vision tools. 💜

Synthetic data generation for tabular data

Efficiently computes derivatives of NumPy code.

Machine learning metrics for distributed, scalable PyTorch applications.

Probabilistic time series modeling in Python

MTEB: State-of-the-art evaluation of embeddings across languages and modalities

State-of-the-Art Embeddings, Retrieval, and Reranking

Simple and Distributed Machine Learning Python Library porting ML algorithms for Spark

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Time series forecasting with PyTorch

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

CUDA Templates and Python DSLs for High-Performance Linear Algebra

Graph Neural Network Library for PyTorch

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

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools, with 10x faster training through evolutionary hyperparameter optimization.

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

⚡️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.

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