yzhwang

jax-multi-gpu-resnet50-example

An example showing how to use jax to train resnet50 on multi-node multi-GPU

moderngpu

Patterns and behaviors for GPU computing

plas

Parallel Linear Algebra Subroutines

tensorflow-wheels

Custom built tensorflow wheels.

zhua

Forum crawler using Python Scrapy

awesome-ai-infrastructures

:tent: Infrastructures™ for Machine Learning Training / Inference in Production.

awesome-machine-learning-in-compilers

Must read research papers and links to tools and datasets that are related to using machine learning for compilers and systems optimisation

bert

TensorFlow code and pre-trained models for BERT

civ-sim

cub

CUB is a flexible library of cooperative threadblock primitives and other utilities for CUDA kernel programming.

cublas-benchmark

Simple benchmark program for cublas routines

Enterprise

Source Code of Enterprise Project @ GWU

gnn

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

gunrock

High-performance Graph Primitives on GPU

instant-ngp

Instant neural graphics primitives: lightning fast NeRF and more

llm.c

LLM training in simple, raw C/CUDA

Medusa

Medusa: Building GPU-based Parallel Sparse Graph Applications with Sequential C/C++ Code

morphic

An AI-powered answer engine with a generative UI

mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

openai-gemm

Open single and half precision gemm implementations

owl

pbrt-v4

Source code to pbrt, the ray tracer described in the forthcoming 4th edition of the "Physically Based Rendering: From Theory to Implementation" book.

plato

腾讯高性能图计算框架Plato

redsync

implementation of redsync

sprocketnes

NES emulator written in Rust

tensorflow

Computation using data flow graphs for scalable machine learning

tf_weights

A TensorFlow implementation of AlexNet with pretrained weights

ThunderKittens

Tile primitives for speedy kernels

yzhwang.github.io

My personal page.

zero123

Zero-1-to-3: Zero-shot One Image to 3D Object: https://zero123.cs.columbia.edu/