adding 'bayesian' layers but need to modify protobufs and vision layers or wtv