Highlights
Stars
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
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.
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
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.