from .ste import STEFunction, StraightThroughEstimator