Add scaffolding for basic hooked method

I want to call hooked(model) to get a hooked model, in the same sense as vmap(model) would get a batched model