Stars
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Fast and memory-efficient exact attention
A retargetable MLIR-based machine learning compiler and runtime toolkit.
A library for efficient similarity search and clustering of dense vectors.
Open standard for machine learning interoperability
Seamless operability between C++11 and Python
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
A platform for Reasoning systems (Reinforcement Learning, Contextual Bandits, etc.)
Riemannian Adaptive Optimization Methods with pytorch optim
Cross-platform, customizable ML solutions for live and streaming media.
open Multi-View Stereo reconstruction library
NumPy aware dynamic Python compiler using LLVM
Open-source simulator for autonomous driving research.
Digital Human Resource: 2D/3D/4D Human Modeling, Avatar Generation & Animation, Clothed People Digitalization, Virtual Try-On, etc.
Graph Neural Network Library for PyTorch
Carbon Language's main repository: documents, design, implementation, and related tools. (NOTE: Carbon Language is experimental; see README)
Refine high-quality datasets and visual AI models
A model-driven approach to building AI agents in just a few lines of code.
The fundamental package for scientific computing with Python.
a language for fast, portable data-parallel computation
Bringing Characters to Life with Computer Brains in Unity
Vector (and Scalar) Quantization, in Pytorch