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

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.

MPPDE.jl

Message Passing Neural PDE Solvers in Julia

generative-ai-for-beginners

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

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

CUDA Templates for Linear Algebra Subroutines

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.

GemmCUDA.jl

Gemm with Tensor Cores

General

The official registry of general Julia packages

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

NeuralPDE.jl

Physics-Informed Neural Networks (PINN) and Deep BSDE Solvers of Differential Equations for Scientific Machine Learning (SciML) accelerated simulation

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