jax_privacy.noise_addition.SupportedStrategies

class jax_privacy.noise_addition.SupportedStrategies(value)[source]

Bases: Enum

Supported strategies for generating intermediate noise.

Attributes

`DEFAULT`	Basic approach for single-machine training scenarios.
`ZERO`	Zero-redundancy approach suitable for multi-machine scenarios.

DEFAULT = (<function SupportedStrategies.<lambda>>, <function SupportedStrategies.<lambda>>): Basic approach for single-machine training scenarios.

ZERO = (<function flatten_with_zero_redundancy>, <function local_reshape_add>): Zero-redundancy approach suitable for multi-machine scenarios. Requires inputs to have explicit sharding annotations.