google / qhbm-library

Quantum Hamiltonian-Based Models built on TensorFlow Quantum
https://qhbm-library.readthedocs.io/en/latest/
Apache License 2.0
40 stars 15 forks source link

Enable EBM seed #143

Closed zaqqwerty closed 2 years ago

zaqqwerty commented 2 years ago

Add seed argument to EnergyInference subclasses.

Part of addressing #85. Follows this TFP guide for usage of seeds with TFP ops. Adding a seed will enable multiple calls to a given sampler to yield the same results, which should end flakiness of tests for #135.