yzhwang

UC Davis PhD from @owensgroup. Co-Founder, CTO at PolyLabs Inc., ex-@sail-sg, ex-@Tencent and Wechat, ex-@google Brain. Built @gunrock. @tensorflow contributor.

PolyLabs Inc.Beijing

Pinned Repositories

gunrock
Programmable CUDA/C++ GPU Graph Analytics
Language:C++952 77 744197
mini
mini is mini
Language:C++17 6 65
hloenv
an environment based on XLA for deep learning compiler optimization research.
Language:C++23 6 41
plato
腾讯高性能分布式图计算框架Plato
Language:C++1.9k 80 135330
bitstarter
Language:JavaScript1 2 00
depixelization
Depixelization of pixel art on GPU
Language:C++22 2 10
jax-multi-gpu-resnet50-example
An example showing how to use jax to train resnet50 on multi-node multi-GPU
Language:Python20 2 01
moderngpu
Patterns and behaviors for GPU computing
Language:C++2 2 02
plas
Parallel Linear Algebra Subroutines
Language:C++1 1 00

yzhwang's Repositories

yzhwang/jax-multi-gpu-resnet50-example
An example showing how to use jax to train resnet50 on multi-node multi-GPU
Language:Python20 2 01
yzhwang/moderngpu
Patterns and behaviors for GPU computing
Language:C++2 2 02
yzhwang/plas
Parallel Linear Algebra Subroutines
Language:C++1 1 00
yzhwang/tensorflow-wheels
Custom built tensorflow wheels.
1 2 1
yzhwang/zhua
Forum crawler using Python Scrapy
Language:Python1 2 01
yzhwang/awesome-ai-infrastructures
:tent: Infrastructures™ for Machine Learning Training / Inference in Production.
1 0
yzhwang/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
1 0
yzhwang/bert
TensorFlow code and pre-trained models for BERT
Language:Python1 0
yzhwang/char-rnn-tensorflow
Multi-layer Recurrent Neural Networks (LSTM, RNN) for character-level language models in Python using Tensorflow
Language:Python2 0
yzhwang/civ-sim
Language:JavaScript
yzhwang/cub
CUB is a flexible library of cooperative threadblock primitives and other utilities for CUDA kernel programming.
Language:Cuda1 0
yzhwang/cublas-benchmark
Simple benchmark program for cublas routines
Language:Cuda1 0
yzhwang/Enterprise
Source Code of Enterprise Project @ GWU
Language:Cuda2 0
yzhwang/gnn
TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.
Language:Python1 0
yzhwang/gunrock
High-performance Graph Primitives on GPU
Language:Cuda
yzhwang/instant-ngp
Instant neural graphics primitives: lightning fast NeRF and more
Language:Cuda0 0
yzhwang/llm.c
LLM training in simple, raw C/CUDA
yzhwang/Medusa
Medusa: Building GPU-based Parallel Sparse Graph Applications with Sequential C/C++ Code
Language:Cuda2 0
yzhwang/morphic
An AI-powered answer engine with a generative UI
Language:TypeScript
yzhwang/mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Language:Python1 0
yzhwang/openai-gemm
Open single and half precision gemm implementations
Language:C1 0
yzhwang/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.
Language:C++1 0
yzhwang/plato
腾讯高性能图计算框架Plato
Language:C++
yzhwang/redsync
implementation of redsync
Language:Cuda1
yzhwang/sprocketnes
NES emulator written in Rust
Language:Rust1 0
yzhwang/tensorflow
Computation using data flow graphs for scalable machine learning
Language:C++1 0
yzhwang/tf_weights
A TensorFlow implementation of AlexNet with pretrained weights
Language:Python1 0
yzhwang/ThunderKittens
Tile primitives for speedy kernels
Language:Cuda
yzhwang/yzhwang.github.io
My personal page.
Language:JavaScript
yzhwang/zero123
Zero-1-to-3: Zero-shot One Image to 3D Object: https://zero123.cs.columbia.edu/
Language:Python