Transforms Module

Function transformations (grad, jit, vmap, etc.)

Modules

• Grad
• Jacfwd
• Jacrev
• Jit
• Jvp
• Vjp
• Vmap
• Utils

Module Overview