Releases: muchlakshay/Forge
Release list
Forge v1.0 - GPT-2 and MNIST demo tests
Windows-x64and Linux-x64 build containing two test executables built on the Forge framework:
-
gpt2.exe - loads real GPT-2 small weights into a GPT-2 architecture built entirely on Forge (custom tensor engine, AVX2 SIMD kernels, OpenBLAS-backed GEMM). Greedy-decoded output matches Hugging Face's transformers implementation token-for-token. Includes an auto-download step for the GPT-2 weights on first run.
-
mnist.exe - MNIST digit classifier demo, also built on Forge. Includes an auto-download step for the MNIST dataset on first run. It trains for 50 epochs, and then print predictions and ground truth
Requirements: a CPU with AVX2 support (standard on any x86-64 machine from the last ~10 years). No other dependencies, both executables are statically linked, and libopenblas.dll is included in the zip.
To run:
- windows: extract the zip, then run
gpt2.exeormnist.exedirectly from
the extracted folder - linux: requires openBLAS installed system-wide (
sudo apt install libopenblas0), in (/usr).