jax

060c8edd - intial jet xform (higher-order taylor series ad)

intial jet xform (higher-order taylor series ad) Co-authored-by: Jesse Bettencourt <jessebett@cs.toronto.edu> Co-authored-by: David Duvenaud <duvenaud@cs.toronto.edu> Co-authored-by: Dougal Maclaurin <dougalm@google.com>