Highlights
- Pro
Stars
The official Python library for the OpenAI API
Neural Networks: Zero to Hero
A latent text-to-image diffusion model
A framework for few-shot evaluation of autoregressive language models.
jax-triton contains integrations between JAX and OpenAI Triton
Graph dump of torchbench models, huggingface models, and TIMM models.
Gin provides a lightweight configuration framework for Python
Library for reading and writing large multi-dimensional arrays.
🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools
MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.
This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our conti…
Solve puzzles. Improve your pytorch.
Development repository for the Triton language and compiler
Language model alignment-focused deep learning curriculum
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
ML Collections is a library of Python Collections designed for ML use cases.
A very fast and expressive template engine.
Beyond the Imitation Game collaborative benchmark for measuring and extrapolating the capabilities of language models
NL-Augmenter 🦎 → 🐍 A Collaborative Repository of Natural Language Transformations
Task-based datasets, preprocessing, and evaluation for sequence models.
This contains the Flax model of min(DALL·E) and code for converting it to PyTorch
Locating and editing factual associations in GPT (NeurIPS 2022)