Skip to content

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

GPT-2 forward pass in raw WGSL (WebGPU)

Resources

Stars

1 star

Watchers

2 watching

Forks

Releases

Packages

Used by

Contributors

Languages