Ethan Wu's repositories
mojo
The Mojo Programming Language
llm.c
LLM training in simple, raw C/CUDA
tinygrad
You like pytorch? You like micrograd? You love tinygrad! ❤️
simulai
A toolkit with data-driven pipelines for physics-informed machine learning.
chibicc
A small C compiler
Lux.jl
Explicitly Parameterized Neural Networks in Julia
ChatGPT-Next-Web
A well-designed cross-platform ChatGPT UI (Web / PWA / Linux / Win / MacOS). 一键拥有你自己的跨平台 ChatGPT 应用。
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.
cuda-samples
Samples for CUDA Developers which demonstrates features in CUDA Toolkit
generative-ai-for-beginners
12 Lessons, Get Started Building with Generative AI 🔗 https://microsoft.github.io/generative-ai-for-beginners/
modulus
A PyTorch based deep-learning toolkit for developing DL models for physical systems
NeuralGraphPDE.jl
Integrating Neural Ordinary Differential Equations, the Method of Lines, and Graph Neural Networks
TensorContractions.jl
Native Tensor Contractions on GPU
DenoisingDiffusion.jl
A pure Julia implementation of denoising diffusion probabilistic models
Chi.jl
Neural Networks Strongly Accelerated on GPUs
GemmCUDA.jl
Gemm with Tensor Cores
FlashAttention.jl
Julia implementation of the Flash Attention algorithm
TaylorDiff.jl
Taylor-mode automatic differentiation for higher-order derivatives
cutlass
CUDA Templates for Linear Algebra Subroutines