🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

提取微信聊天记录，将其导出成HTML、Word、Excel文档永久保存，对聊天记录进行分析生成年度聊天报告，用聊天数据训练专属于个人的AI聊天助手

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

The official Python library for the OpenAI API

A generative world for general-purpose robotics & embodied AI learning.

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Hydra is a framework for elegantly configuring complex applications

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

Use PEFT or Full-parameter to finetune 450+ LLMs (Qwen2.5, InternLM3, GLM4, Llama3.3, Mistral, Yi1.5, Baichuan2, DeepSeek-R1, ...) and 150+ MLLMs (Qwen2.5-VL, Qwen2-Audio, Llama3.2-Vision, Llava, I…

Denoising Diffusion Probabilistic Models

[CVPR 2023 Best Paper Award] Planning-oriented Autonomous Driving

[ECCV 2022] This is the official implementation of BEVFormer, a camera-only framework for autonomous driving perception, e.g., 3D object detection and semantic map segmentation.

Collection of reinforcement learning algorithms

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 Spotlight.

A collection of reference environments for offline reinforcement learning

⚡ InstaFlow! One-Step Stable Diffusion with Rectified Flow (ICLR 2024)

A flexible framework powered by ComfyUI for generating personalized Nobel Prize images.

[NeurIPS 2024 Datasets and Benchmarks Track] Closed-Loop E2E-AD Benchmark Enhanced by World Model RL Expert

Official Implementation of Rectified Flow (ICLR2023 Spotlight)

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations

Official implementation of Cold-Diffusion for different transformations in pytorch.

Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"

A JAX-based simulator for autonomous driving research.