Open source code for AlphaFold.

A Locality Sensitive Hashing (LSH) library with an emphasis on large, highly-dimensional datasets.

2D and 3D molecular visualization in Jupyter notebooks using 3DMol.js and D3.js

PyPi module for Graphlet AI Knowledge Graph Factory

JAX-based neural network library

openpilot is an operating system for robotics. Currently, it upgrades the driver assistance system in 275+ supported cars.

Graph Neural Network Library for PyTorch

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

PyGraphistry is a Python library to quickly load, shape, embed, and explore big graphs with the GPU-accelerated Graphistry visual graph analyzer

Reference implementation of the Transformer architecture optimized for Apple Neural Engine (ANE)

Minimal code to train a Large Language Model (LLM).

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)

Sparse learning library and sparse momentum resources.

The comma.ai Calibration Challenge!

Simple Annotated implementation of GPT-NeoX in PyTorch

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

Code release for NeRF (Neural Radiance Fields)

Memory-efficient transformer. Work in progress.

Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch

Automatic mixed precision training and inference guide for GPU

RoBERTa with Flax

Baby deep learning library🐣

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

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

100 Must-Read NLP Papers