Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Ongoing research training transformer models at scale
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.
SkyPilot: Run AI and batch jobs on any infra (Kubernetes or 12+ clouds). Get unified execution, cost savings, and high GPU availability via a simple interface.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Library for reading and writing large multi-dimensional arrays.
Flax is a neural network library for JAX that is designed for flexibility.
A flexible, high-performance 3D simulator for Embodied AI research.
Open-source simulator for autonomous driving research.
The official Python library for the OpenAI API
Examples and guides for using the OpenAI API
A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
A framework for few-shot evaluation of language models.
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
Models and examples built with TensorFlow
A feature-rich command-line audio/video downloader
LLM training code for Databricks foundation models
jax-triton contains integrations between JAX and OpenAI Triton
Python library for loading and using triangular meshes.
RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference,…
A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)
Fast and robust date extraction from web pages, with Python or on the command-line
Text-to-video generation: CogVideoX (2024) and CogVideo (ICLR 2023)
Unsupervised text tokenizer for Neural Network-based text generation.