Multi-Layer Perceptron
Physics-Informed Neural Networks (PINNs)
DeepONet: Operator Learning with Neural Networks
Differentiable Simulation
Graph Neural Network
Lab Setup

JAX Transformations: The Power of Composition

JAX's key advantage is that transformations compose. You can combine grad, jit, and vmap in any order:

Back
 
Next
©  |   Cornell University    |   Center for Advanced Computing    |   Copyright Statement    |   Access Statement
CVW material development is supported by NSF OAC awards 1854828, 2321040, 2323116 (UT Austin) and 2005506 (Indiana University)