Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A machine learning compiler for GPUs, CPUs, and ML accelerators
π€ Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
Refine high-quality datasets and visual AI models
Autonomous coding agent right in your IDE, capable of creating/editing files, executing commands, using the browser, and more with your permission every step of the way.
AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Development repository for the Triton language and compiler
Data manipulation and transformation for audio signal processing, powered by PyTorch
A Flexible Framework for Experiencing Cutting-edge LLM Inference Optimizations
Train transformer language models with reinforcement learning.
Composable building blocks to build Llama Apps
PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
π¦π The platform for reliable agents.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
One-for-All Multimodal Evaluation Toolkit Across Text, Image, Video, and Audio Tasks
A library for efficient similarity search and clustering of dense vectors.
Cross-platform, customizable ML solutions for live and streaming media.
Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages
LlamaIndex is the leading framework for building LLM-powered agents over your data.
βοΈ π π π Evaluating state of the art in AI
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (Vβ¦
Examples and guides for using the OpenAI API