#pytorch

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

Accessible large language models via k-bit quantization for PyTorch.

A retargetable MLIR-based machine learning compiler and runtime toolkit.

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

Bayesian optimization in PyTorch

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

We write your reusable computer vision tools. 💜

sbi is a Python package for simulation-based inference, designed to meet the needs of both researchers and practitioners. Whether you need fine-grained control or an easy-to-use interface, sbi has you covered.

Transformers-compatible library for applying various compression algorithms to LLMs for optimized deployment with vLLM

Machine learning metrics for distributed, scalable PyTorch applications.

Probabilistic time series modeling in Python

Machine learning on FPGAs using HLS

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

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

Time series forecasting with PyTorch

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization pipelines, and OpenAI-compatible/MCP serving.

Open security scanner and self-hosted control plane for AI, MCP, and cloud. One evidence model — run scans in your environment, centralize findings, govern in your VPC.

Graph Neural Network Library for PyTorch

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

Unsloth is a local UI for training and running Kimi K3, Gemma 4, Qwen3.6, DeepSeek-V4, GLM and other models.

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

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

Emulate simulations easily

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT