pytorch

PyTorch implementations for the Bitfount framework.

Module

bitfount.backends.pytorch.data - Data components for the PyTorch backend.
bitfount.backends.pytorch.epoch_callbacks - Pytorch Lightning Callbacks to trigger Bitfount Epoch Hooks.
bitfount.backends.pytorch.federated - Federated components for the PyTorch backend.
bitfount.backends.pytorch.loss - Implements loss for pytorch modules.
bitfount.backends.pytorch.models - Models utilising the PyTorch backend.
bitfount.backends.pytorch.types - Type Variable for our PyTorch Models.
bitfount.backends.pytorch.utils - Contains PyTorch specific utility methods.
bitfount.backends.pytorch.weight_clipper - Parameter clipping for continuous attributes secure aggergation.