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

Robust Speech Recognition via Large-Scale Weak Supervision

Deep Learning for humans

Inference code for Llama models

The simplest, fastest repository for training/finetuning medium-sized GPTs.

TensorFlow code and pre-trained models for BERT

OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamically to do so.

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

Reverse engineered ChatGPT API

[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.

Graph Neural Network Library for PyTorch

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Best Practices on Recommendation Systems

Magenta: Music and Art Generation with Machine Intelligence

State-of-the-Art Text Embeddings

Open source code for AlphaFold 2.

An open source implementation of CLIP.

Run any open-source LLMs, such as DeepSeek and Llama, as OpenAI compatible API endpoint in the cloud.

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.

Running large language models on a single GPU for throughput-oriented scenarios.

A collection of libraries to optimise AI model performances

Matplotlib styles for scientific plotting

Automated Machine Learning with scikit-learn

DoWhy is a Python library for causal inference that supports explicit modeling and testing of causal assumptions. DoWhy is based on a unified language for causal inference, combining causal graphic…

Implementation of Graph Convolutional Networks in TensorFlow

Code release for ConvNeXt model

The GitHub repository for the paper "Informer" accepted by AAAI 2021.

Uplift modeling and causal inference with machine learning algorithms