Skip to content

loristns/wgt

Repository files navigation

WGT

WGT is a tiny implementation of a minimal deep learning inference runtime for the browser, through WebGPU.

It's written in TypeScript, and WGSL (WebGPU Shading Language) with the goal of being as readable as possible.

For now, it only implements the ops needed for the forward pass of GPT-2.

This is very much a work in progress, expect rough edges, unoptimized code, bugs, and missing features.

Usage

npm install
npm run dev

About

Minimal WebGPU inference runtime

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published