Skip to content

Releases: muchlakshay/Forge

Forge v1.0 - GPT-2 and MNIST demo tests

Choose a tag to compare

@muchlakshay muchlakshay released this 16 Aug 17:37

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.exe or mnist.exe directly from
    the extracted folder
  • linux: requires openBLAS installed system-wide (sudo apt install libopenblas0), in (/usr).