sigma_repram_layers.py contains a PyTorch implementation of two layers: SRLinear and SRConv2d. These two layers can be used in palce of the regular torch.nn.Linear and torch.nn.Conv2d layers in a plug and play fashion. 