Pinned Repositories
atex
A TensorFlow Extension: GPU performance tools for TensorFlow.
cmake-examples
Useful CMake Examples
dotfiles
flax
Flax is a neural network library for JAX that is designed for flexibility.
minGPT
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
out-of-the-box-fp8-training
Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
scripts
tensorflow
An Open Source Machine Learning Framework for Everyone
wenchenvincent's Repositories
wenchenvincent/atex
A TensorFlow Extension: GPU performance tools for TensorFlow.
wenchenvincent/cmake-examples
Useful CMake Examples
wenchenvincent/dotfiles
wenchenvincent/flax
Flax is a neural network library for JAX that is designed for flexibility.
wenchenvincent/minGPT
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
wenchenvincent/nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
wenchenvincent/out-of-the-box-fp8-training
Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
wenchenvincent/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
wenchenvincent/scripts
wenchenvincent/tensorflow
An Open Source Machine Learning Framework for Everyone
wenchenvincent/TransformerEngine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in both training and inference.
wenchenvincent/triton
Development repository for the Triton language and compiler
wenchenvincent/unit-scaling
A library for unit scaling in PyTorch
wenchenvincent/unit-scaling-demo
Unit Scaling demo and experimentation code
wenchenvincent/wenchenvincent.github.io
wenchenvincent/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators