Bringing PDEs to JAX with forward and reverse modes automatic differentiation.

Published in: CoRR (2023)

Keyphrases

partial differential equations
fully automatic
numerical solution
computer vision
neural network
genetic algorithm
expert systems
diffusion equation
database
natural images
image quality
data driven
level set
social networks
machine learning
real world
real time