* fix(mlx-lm): the top p imp * chore: address comment
* chore(mlx-lm): clean up the top p imp * chore: clean up * chore: add test * chore: address comments * chore: clean up docs string * chore: clean up test