How to dispatch?

[dirmeier]

Not sure which is better

__call__(self, method="predict", **kwargs):

or dispatch based on number of args:

__call__(self, x **kwargs):
  if "y" in kwargs:
      return self._this_method(x, kwargs["y"]):
  return return self._that_method(x)   

Check what Haiku/Flax/.. recommend.