Documentation
¶
Overview ¶
GPU Training Example — demonstrates Born GPU training loop.
A 3-layer MLP trained on synthetic classification data. Validates that GPU training works end-to-end: forward → backward → optimizer → repeat.
Usage:
go run ./examples/gpu-training go run ./examples/gpu-training -steps 50 -batch 128 -hidden 512
Source Files
Click to show internal directories.
Click to hide internal directories.