voiage.main_backends
Computational backend dispatch and capability reporting.
Classes
Section titled “Classes”- TrialArmProtocol Minimal trial arm interface required by the backend helpers.
- TrialDesignProtocol Minimal trial design interface required by the backend helpers.
- ParameterSetProtocol Minimal parameter-set interface required by the backend helpers.
- ModelOutputProtocol Minimal model output interface required by the backend helpers.
- ModelFuncProtocol Callable contract for model functions used by EVSI helpers.
- Backend Abstract base class for computational backends.
- NumpyBackend NumPy-based computational backend.
- JaxBackend Public JAX backend type.
- AppleMetalBackend Optional Apple Metal backend backed by PyTorch MPS.
Functions
Section titled “Functions”- get_backend Get a computational backend by name.
- set_backend Set the default computational backend.
- benchmark_evpi Measure a backend’s EVPI throughput on a fixed workload.
- benchmark_memory_throughput Measure a backend using the memory/throughput benchmark shape.
- benchmark_mps_vs_cpu Compare a CPU benchmark payload against an optional Apple Metal payload.
- compile_phase_3_handoff_packet Build a deterministic Phase-3 review packet for both scalar and memory workloads.