pytorch GitHub repositories

fastmachinelearning/hls4ml

Machine learning on FPGAs using HLS

★ 2.2K+1 todayPython

kitops-ml/kitops

An open source DevOps tool from the CNCF for packaging and versioning AI/ML models, datasets, code, and configuration into an OCI Artifact.

dusty-nv/jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

★ 4.9K+1 todayJupyter Notebook

girafe-ai/ml-course

Open Machine Learning course

★ 3.6K+1 todayJupyter Notebook

machine-learning python

awslabs/gluonts

Probabilistic time series modeling in Python

★ 5.3K+1 todayPython

mlmed/torchxrayvision

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

pytorch/TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

★ 3.0K+1 todayPython

MaximeVandegar/Papers-in-100-Lines-of-Code

Implementation of papers in 100 lines of code.

thu-pacman/chitu

High-performance inference framework for large language models, focusing on efficiency, flexibility, and availability.

★ 3.0K+1 todayPython

microsoft/Tutel

Tutel MoE: Optimized Mixture-of-Experts Library, Support GLM/Kimi/Gpt/DeepSeek/Qwen using FP8/NVFP4/MXFP4

★ 1.0K+1 todayC

tomas-gajarsky/facetorch

Python library for analysing faces using PyTorch

lukasmasuch/best-of-ml-python

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

TorchIO-project/torchio

Medical imaging processing for AI applications.

★ 2.5K+1 todayPython

huggingface/peft

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

★ 21.9K+0 todayPython

deepspeedai/DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

★ 43.3K+0 todayPython

huggingface/datasets

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

★ 22.1K+0 todayPython

llm ai

Exorust/TorchLeet

LeetCode for PyTorch — 65 ML/AI interview problems from real interviews at Google, Meta, Anthropic. Jupyter notebooks, an auto-grader, and an MCP AI tutor.

★ 2.6K+0 todayJupyter Notebook

llm deep-learning

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

★ 4.3K+0 todayPython

llm machine-learning

kornia/kornia

🐍 Geometric Computer Vision Library for Spatial AI

★ 11.4K+0 todayPython

pytorch/rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

★ 3.6K+0 todayPython

ai machine-learning

Lightning-AI/pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

★ 31.4K+0 todayPython

ai machine-learning

Nixtla/neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

NVIDIA/DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

★ 5.8K+0 todayC++

machine-learning python

ndif-team/nnsight

The nnsight package enables interpreting and manipulating the internals of deep learned models.

★ 1.1K+0 todayPython

zenml-io/zenml

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

NVIDIA-BioNeMo/bionemo-recipes

BioNeMo Recipes: For building and adapting AI models in drug discovery at scale

★ 878+0 todayPython

open-edge-platform/geti

Build, train, optimize, and run computer vision models locally, from raw images to live inference. Open source, optimized for Intel® XPU (CUDA also supported).

★ 1.4K+0 todayPython

wandb/wandb

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

arc53/DocsGPT

Private AI platform for agents, assistants and enterprise search. Built-in Agent Builder, Deep research, Document analysis, Multi-model support, and API connectivity for agents.

jeshraghian/snntorch

Deep and online learning with spiking neural networks in Python

★ 2.1K+0 todayPython

WenjieDu/PyPOTS

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

★ 2.1K+0 todayPython

NVIDIA/kvpress

LLM KV cache compression made easy

★ 1.2K+0 todayPython

pyro-ppl/pyro

Deep universal probabilistic programming with Python and PyTorch

★ 9.1K+0 todayPython

whitecircle/halo

Halo is an open-source framework built by White Circle for training large language and multimodal models

EnzymeAD/Enzyme

High-performance automatic differentiation of LLVM and MLIR.

charliedream1/ai_wiki

《AI全栈-全网优秀资源搜集站》：搜集全网优秀资源，记载工程实践问题的解决策略与关键要点，分享各种实用案例，追踪前沿技术发展，囊括 AI 全栈知识，涵盖大模型、编程技术、机器学习、深度学习、强化学习、图神经网络、语音识别、NLP 及图像识别等领域

★ 747+0 todayJupyter Notebook

llm ai

PennyLaneAI/pennylane

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

★ 3.5K+0 todayPython

TingsongYu/PyTorch-Tutorial-2nd

《Pytorch实用教程》（第二版）无论是零基础入门，还是CV、NLP、LLM项目应用，或是进阶工程化部署落地，在这里都有。相信在本书的帮助下，读者将能够轻松掌握 PyTorch 的使用，成为一名优秀的深度学习工程师。

★ 4.6K+0 todayJupyter Notebook

Ramakm/AI-Engineering

A group of notebooks and other files which can help you learn AI from scratch.

★ 1.5K+0 todayJupyter Notebook

ai machine-learning

pytorch/audio

Data manipulation and transformation for audio signal processing, powered by PyTorch

★ 3.0K+0 todayPython

sbi-dev/sbi

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.

★ 879+0 todayPython