Hi I have noticed the output in the BNN tutorial is erroring out as soon as `StatefulLuxLayer` is introduced:  This impacts v0.34. Regards Patrick