Tensors and Dynamic neural networks in Python with strong GPU acceleration

为GPT/GLM等LLM大语言模型提供实用化交互接口，特别优化论文阅读/润色/写作体验，模块化设计，支持自定义快捷按钮&函数插件，支持Python和C++等项目剖析&自译解功能，PDF/LaTex论文翻译&总结功能，支持并行问询多种LLM模型，支持chatglm3等本地模型。接入通义千问, deepseekcoder, 讯飞星火, 文心一言, llama2, rwkv, claude2, m…

Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

OpenMMLab Detection Toolbox and Benchmark

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Open standard for machine learning interoperability

PyTorch implementations of Generative Adversarial Networks.

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

End-to-End Object Detection with Transformers

Mamba SSM architecture

pix2tex: Using a ViT to convert images of equations into LaTeX code.

A one-stop, open-source, high-quality data extraction tool, supports PDF/webpage/e-book extraction.一站式开源高质量数据提取工具，支持PDF/网页/多格式电子书提取。

An educational resource to help anyone learn deep reinforcement learning.

Geometric Computer Vision Library for Spatial AI

The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

The most widely used Python to C compiler

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

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

Deep universal probabilistic programming with Python and PyTorch

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

Pytorch implementation of convolutional neural network visualization techniques

Matplotlib styles for scientific plotting

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

Image augmentation library in Python for machine learning.