Skip to content

Developer Guide

Everything you need to contribute to TorchEBM. Pick the page that matches the task at hand.

  • Getting Started


    Clone, install, branch, commit (Conventional Commits), PR.

  • Code Guidelines


    Style, API design, performance rules, testing conventions.

  • Architecture


    Package layout and how the core abstractions compose.

  • Performance


    Vectorisation, device / memory patterns, mixed precision, sampler tuning.

  • Benchmarking


    benchmarks/run.sh. wall-time regression tracking and the dashboard.

  • Profiling


    benchmarks/profiler.py. per-op CPU/CUDA drill-down and A/B diffs.