Can guide and model functions be class member?

I wanna pack relevant functions into class, as bound member(belongto instance and can accept “self” parameter)

Yes, I always make my code in the form of:

class ProbModule:
  def __init__(self, f1, f2):
     # assignments etc.
    self.f1 = f1
    self.f2 = f2
  def model(self, *args):
     ...
  def guide(self, *args):
    ...
  def infer(self, *args):
    ...