jax

b0b88d87 - [attrs] add linearize and vjp support