Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
Simplify your onnx model
LightLLM is a Python-based LLM (Large Language Model) inference and serving framework, notable for its lightweight design, easy scalability, and high-speed performance.
β‘οΈ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.
NeuralProphet: A simple forecasting package
TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data
A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.
A python library for self-supervised learning on images.
Synthetic data generation for tabular data
Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Multilingual Automatic Speech Recognition with word-level timestamps and confidence
AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.
Pytorch domain library for recommendation systems
Deep learning PyTorch library for time series forecasting, classification, and anomaly detection (originally for flood forecasting).
A sketch extractor for anime/illustration.
A Python toolkit/library for reality-centric machine/deep learning & data mining on partially-observed time series, with 50+ SOTA neural network models for scientific analysis tasks (imputation, classification, clustering, forecasting, anomaly detection, cleaning) on incomplete industrial irregularly-sampled multivariate TS with NaN missing values
YomiTokuγ―AIγζ΄»η¨γγζ₯ζ¬θͺζζΈθ§£ζγ¨γ³γΈγ³γζδΎγγPythonγγγ±γΌγΈγ§γγ Yomitoku is an AI-powered document image analysis package designed specifically for the Japanese language.
Python library for time series forecasting using scikit-learn compatible models, statistical methods, and foundation models
AuK: An Open-Source Foundational Model for Speech Generation and Editing
Computations and statistics on manifolds with geometric structures.
A toolkit for time series machine learning and deep learning
Automated Time Series Forecasting
TabICLv2: An open tabular foundation model
Toolkit to segment text into sentences or other semantic units in a robust, efficient and adaptable way.
Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
Official PyTorch implementation of Superpoint Transformer [ICCV'23], SuperCluster [3DV'24 Oral], and EZ-SP [ICRA'26]
Formula Recognition & Office Editing | Handwriting & PDF to LaTeX/Markdown | Secure API Integrations.
cuDNN Frontend is NVIDIA's modern, open-source entry point to the cuDNN library and a growing collection of high-performance open-source kernels.
LibreYOLO is a MIT licensed open source computer vision library
High-performance Vision library in Python. Scale your research, not boilerplate.
Resources about time series forecasting and deep learning.
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
The official Roboflow Python package. Manage your datasets, models, and deployments. Roboflow has everything you need to build a computer vision application.
Image augmentation for computer vision. AGPL-3.0-only or commercial licensing.
This repo is a alternative solution for standard self-attention. It uses tree routing with learnable functions for cheaper self-attention while not compromising accuracy.
π« Industrial-strength Natural Language Processing (NLP) in Python
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Graph Neural Network Library for PyTorch
100-Days-Of-ML-CodeδΈζη
Best Practices on Recommendation Systems
Open standard for machine learning interoperability
RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RNN and transformer - great performance, linear time, constant space (no kv-cache), fast training, infinite ctx_len, and free sentence embedding.
Open Machine Learning Compiler Framework
Low-code framework for building custom LLMs, neural networks, and other AI models
π Geometric Computer Vision Library for Spatial AI
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
Minimal and clean examples of machine learning algorithms implementations
The Triton Inference Server provides an optimized cloud and edge inferencing solution.
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection with training, validation, inference, and multi-format export.
A Python library for anomaly detection across tabular, time series, graph, text, image, and audio data. 60+ detectors, benchmark-backed ADEngine orchestration, and an agentic workflow for AI agents.
Perform data science on data that remains in someone else's server
End-to-End Speech Processing Toolkit
TensorFlow-based neural network library
so-vits-svc fork with realtime support, improved interface and more features.
ModelScope: bring the notion of Model-as-a-Service to life.
Deep universal probabilistic programming with Python and PyTorch
A collection of machine learning examples and tutorials.
BoxMOT: Pluggable Python and C++ SOTA multi-object tracking modules with support for axis-aligned and oriented bounding boxes