quantax.model.SchmittNet

quantax.model.SchmittNet(depth: int, channels: int, kernel_size: int, dtype: ~numpy.dtype = <class 'jax.numpy.complex64'>)

CNN defined in PRL 125, 100503.

Parameters:

• depth – The depth of the network.

• channels – The number of channels. Each layer has the same amount of channels.

• kernel_size – The kernel size. Each layer has the same kernel size.

• dtype – The data type of the parameters.