Lists (17)
Starred repositories
An IOS app for taking survey data of underwater caves.
The python based pyGRACE contains functions for pre processing of CSR GRACE and GRACE-FO for Google Earth Engine (GEE) ingestion. Subsequently, application module will be added for applications lik…
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance…
Dora is an experiment management framework. It expresses grid searches as pure python files as part of your repo. It identifies experiments with a unique hash signature. Scale up to hundreds of exp…
A compact implementation of SGLang, designed to demystify the complexities of modern LLM serving systems.
GCM Physics written in JAX
🚀 MassGen is an open-source multi-agent scaling system that runs in your terminal, autonomously orchestrating frontier models and agents to collaborate, reason, and produce high-quality results. | …
Optimize Julia Functions With MLIR and XLA for High-Performance Execution on CPU, GPU, TPU and more.
Cuq: A MIR-to-Coq Framework Targeting PTX for Formal Semantics and Verified Translation of Rust GPU Kernels
A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its variants as the primary backbone with support for ImageNet train…
A Python package designed for research in diffusion-based generative modeling
This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".
DeepEarth: AI Foundation Model for Planetary Science & Sustainability
An exabyte-scale, multi-region distributed file system
A curated list of global electrical grid maps, datasets and resources.
NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer
Examples demonstrating available options to program multiple GPUs in a single node or a cluster
A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.