sbhavani

keras

Deep Learning for humans

FlexGen

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

TensorRT-LLM

TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT-LLM also contains components to create Python and C++ runtimes that execute those TensorRT engines.

torchtune

A Native-PyTorch Library for LLM Fine-tuning

esm

Evolutionary Scale Modeling (esm): Pretrained language models for proteins

EasyLM

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

RFdiffusion

Code for running RFdiffusion

maxtext

A simple, performant and scalable Jax LLM!

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

hypernerf

Code for "HyperNeRF: A Higher-Dimensional Representation for Topologically Varying Neural Radiance Fields".

levanter

Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax

paxml

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.

nucleotide-transformer

🧬 Nucleotide Transformer: Building and Evaluating Robust Foundation Models for Human Genomics

ProDy

A Python Package for Protein Dynamics Analysis

putting-nerf-on-a-diet

Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis Implementation

Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

zero-bubble-pipeline-parallelism

Zero Bubble Pipeline Parallelism

JAX-Toolbox

JAX-Toolbox

flash-attention-jax

Implementation of Flash Attention in Jax

XLB

XLB: Accelerated Lattice Boltzmann (XLB) based on JAX for Physics-based ML

VaxNeRF

Official implementation of VaxNeRF (Voxel-Accelearated NeRF).

github-trending

使用 Github Actions 跟踪 Github 趋势项目。

microxcaling

PyTorch emulation library for Microscaling (MX)-compatible data formats

iree-nvgpu

JORA

JORA: JAX Tensor-Parallel LoRA Library

nanoGPT-fp8

SwinMTL

mamba-jax

drinkmixer