jax
35e5e644 - make custom_transforms handle pytrees, add api.defvjp

Commit

6 years ago

make custom_transforms handle pytrees, add api.defvjp With advice from @dougalm!

References

#818 - make custom_transforms handle pytrees, add api.defvjp

Author

mattjj

mattjj

Committer

mattjj

mattjj

Parents

Loading