voiage.main_backends.AppleMetalBackend

Optional Apple Metal backend backed by PyTorch MPS.

Methods

evpi

evpi([positional or keyword] self: None = None, [positional or keyword] net_benefit_array: ArrayLike = None) -> float

Calculate EVPI on the Apple Metal device.

Parameters:

• self
• net_benefit_array ArrayLike

Returns: float

enbs_simple

enbs_simple([positional or keyword] self: None = None, [positional or keyword] net_benefit_array: ArrayLike = None, [positional or keyword] research_cost: ArrayLike = None) -> float

Return a legacy clipped EVPI-minus-cost screening score, not COSS.

Parameters:

• self
• net_benefit_array ArrayLike
• research_cost ArrayLike

Returns: float

enbs_simple_jit

enbs_simple_jit([positional or keyword] self: None = None, [positional or keyword] net_benefit_array: ArrayLike = None, [positional or keyword] research_cost: ArrayLike = None) -> float

PyTorch MPS backend does not provide a separate JIT path here.

Parameters:

• self
• net_benefit_array ArrayLike
• research_cost ArrayLike

Returns: float