Reimplement JAX Scan dispatcher with MIT-MOT support

Co-authored-by: Jesse Grabowski <48652735+jessegrabowski@users.noreply.github.com>

Author: ricardoV94