Refactor the JAX utilities for the multinomial model (#18)

* Sketch the refactoring

* Add quasiloglikelihood

* Refactor _add_variant

* Remove deprecated notebook

* Remove redundant imports

* Mark old API constructs as deprecated.

* Replace deprecated functions with new API

* Rename the submodule