YichengDWu

Sophon.jl

Efficient, Accurate, and Streamlined Training of Physics-Informed Neural Networks

MoYe.jl

Programming Gemm Kernels on NVIDIA GPUs with Tensor Cores in Julia

FlashAttention.jl

Julia implementation of the Flash Attention algorithm

NeuralGraphPDE.jl

Integrating Neural Ordinary Differential Equations, the Method of Lines, and Graph Neural Networks

matmul.mojo

High Performance Matrix Multiplication in Pure Mojo 🔥

yoho

A compiler written in Mojo 🔥 and generates RISC-V assembly

tinyedm

Modular PyTorch (Lightning) implementation of Analyzing and Improving the Training Dynamics of Diffusion Models.

generative-ai-for-beginners

12 Lessons, Get Started Building with Generative AI 🔗 https://microsoft.github.io/generative-ai-for-beginners/

random123

Splittable pseudorandom number generators

ChatGPT-Next-Web

A well-designed cross-platform ChatGPT UI (Web / PWA / Linux / Win / MacOS). 一键拥有你自己的跨平台 ChatGPT 应用。

Chi.jl

Neural Networks Strongly Accelerated on GPUs

chibicc

A small C compiler

cuda-samples

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

cutlass-kernels

DenoisingDiffusion.jl

A pure Julia implementation of denoising diffusion probabilistic models

dyffusion

[NeurIPS 2023] A Dynamics-informed Diffusion Model for Spatiotemporal Forecasting

earth2mip

Earth-2 Model Intercomparison Project (MIP) is a python framework that enables climate researchers and scientists to explore and experiment with AI models for weather and climate.

ell

A language model programming framework.

GemmCUDA.jl

Gemm with Tensor Cores

llm.c

LLM training in simple, raw C/CUDA

Lux.jl

Explicitly Parameterized Neural Networks in Julia

mini_edm

modulus

A PyTorch based deep-learning toolkit for developing DL models for physical systems

mojo

The Mojo Programming Language

Mojo-Marathons

simulai

A toolkit with data-driven pipelines for physics-informed machine learning.

TaylorDiff.jl

Taylor-mode automatic differentiation for higher-order derivatives

TensorContractions.jl

Native Tensor Contractions on GPU

tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

yichengdwu.github.io