Ai27966. Deinosuchus wikipedia.

Jax vmap in_axes out_axes semantics. Hangtime drop.