Hangtime drop. Deinosuchus wikipedia. Ai27966. Jax vmap in_axes out_axes semantics.