skip to content
[readonly]

3 packages match "gpt2x"

  • total 4
  • drwxr-xr-x-../
  • drwxr-xr-x-Oct 27 03:26@ajax/mesh-prog-gpt2x/# 0.1.0 gpt2x, GPT-2 124M's exact-mode GPU kernels (embed, norm, GEMM/GEMV, KV, attention, logits) bitwise equal to the CPU reference, as PTX from C
  • drwxr-xr-x-Oct 27 01:45@ajax/mesh-abi-view-gpt2x/# 0.1.0 programs/gpt2x's view: GPT-2 exact-mode kernels' parameter block, ops and epilogues
  • drwxr-xr-x-Oct 27 01:59@ajax/mesh-ref-gpt2/# 0.1.0 GPT-2 124M in the cluster float mode on the CPU: the forward pass, prefill and decode, greedy generation, bitwise equal to the GPU kernels