训练框架

EmotiVoice 😊: a Multi-Voice and Prompt-Controlled TTS Engine

BoxMOT: Pluggable Python and C++ SOTA multi-object tracking modules with support for axis-aligned and oriented bounding boxes

BertViz: Visualize Attention in Transformer Models

Leveraging BERT and c-TF-IDF to create easily interpretable topics.

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

Open Lakehouse Format for Multimodal AI. Convert from Parquet in 2 lines of code for 100x faster random access, vector index, and data versioning. Compatible with Pandas, DuckDB, Polars, Pyarrow, and PyTorch with more integrations coming..

Python & Command-line tool to gather text and metadata on the Web: Crawling, scraping, extraction, output as CSV, JSON, HTML, MD, TXT, XML

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

Open-source offline translation library written in Python

Facebook AI Research's Automatic Speech Recognition Toolkit

This repository contains a hand-curated resources for Prompt Engineering with a focus on Generative Pre-trained Transformer (GPT), ChatGPT, PaLM etc

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

Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.

Fault-tolerant, highly scalable GPU orchestration, and a machine learning framework designed for training models with billions to trillions of parameters

Setup and customize deep learning environment in seconds.

Learn AI and LLMs from scratch using free resources

AI算法岗求职攻略（涵盖准备攻略、刷题指南、内推和AI公司清单等资料）

Transforms PDF, Documents and Images into Enriched Structured Data

Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.

A concise but complete full-attention transformer with a set of promising experimental features from various papers

A PyTorch native platform for training generative AI models

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

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

computer vision and sports

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡

:sunglasses: A curated list of awesome MLOps tools

深度学习辅助漫画翻译工具, 支持一键机翻和简单的图像/文本编辑 | Yet another computer-aided comic/manga translation tool powered by deeplearning

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

A machine learning software for extracting information from scholarly documents

Keras model to generate HTML code from hand-drawn website mockups. Implements an image captioning architecture to drawn source images.

Curated list of the best truly open-source AI projects, models, tools, and infrastructure. Daily updated.

🌟100+ 原创 LLM / RL 原理图📚，《大模型算法》作者巨献！💥（100+ LLM/RL Algorithm Maps ）

Repo for BenCao [original name: HuaTuo (华驼)], Instruction-tuning Large Language Models with Chinese Medical Knowledge. 本草（原名：华驼）模型仓库，基于中文医学知识的大语言模型指令微调

Machine Learning and Agentic AI Resources, Practice and Research

Fast inference engine for Transformer models

Prompt Engineering | Prompt Versioning | Use GPT or other prompt based models to get structured output. Join our discord for Prompt-Engineering, LLMs and other latest research

Lightning fast C++/CUDA neural network framework

[EMNLP 2022] An Open Toolkit for Knowledge Graph Extraction and Construction

Simplify your onnx model

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

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

LightLLM is a Python-based LLM (Large Language Model) inference and serving framework, notable for its lightweight design, easy scalability, and high-speed performance.

The Hugging Face course on Transformers

Serve, optimize and scale PyTorch models in production

Transformers for Information Retrieval, Text Classification, NER, QA, Language Modelling, Language Generation, T5, Multi-Modal, and Conversational AI

AdalFlow: The library to build & auto-optimize LLM applications.

A Python package for segmenting geospatial data with the Segment Anything Model (SAM)

Everything you need to know to build your own RAG application

State of the Art Natural Language Processing

FEDML - The unified and scalable ML library for large-scale distributed training, model serving, and federated learning. FEDML Launch, a cross-cloud scheduler, further enables running any AI jobs on any GPU cloud or on-premise cluster. Built on this library, TensorOpera AI (https://TensorOpera.ai) is your generative AI platform at scale.