FlashMLA: Efficient MLA decoding kernels

DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling

DeepEP: an efficient expert-parallel communication library

Open(MM)DiT: An Easy, Fast and Memory-Efficient System for (MM)DiT Training and Inference

Fully open reproduction of DeepSeek-R1

Export and Backup Ollama models into GGUF and ModelFile

LLM inference in C/C++

Analogue of the scale-invariant feature transform (SIFT) for three-dimensional images. Includes an image processing and linear algebra library with feature matching and RANSAC regression. Also incl…

Global dataset of gridded population and GDP (1980-2010 estimations and 2020-2100 scenarios)

spoof mdoc files from tomo5 frames directories

Coarse-grained molecular dynamics for protein physics. NOTE: this has been superseded by Upside 2: https://github.com/sosnicklab/upside2-md

Neural networks to analyze structural heterogeneity in cryo-electron sub-tomograms

On-Policy Policy Gradient Algorithms in JAX

EDM2 and Autoguidance -- Official PyTorch implementation

Implementation of Video Diffusion Models, Jonathan Ho's new paper extending DDPMs to Video Generation - in Pytorch

Inference Llama 2 in one file of pure C

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

LLM training in simple, raw C/CUDA

Grok open release

Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"

Official release of the ProGen models

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

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

Differentiable signal processing on the sphere for PyTorch

Plugin to use OPUS-DSD programs within the Scipion framework