Comment by skainswoo

vmap is by far my favorite feature in JAX. Very cool to see that you can get the same thing straight out of haskell.