jaxprof.py

This script is used to compare performance of JAX sparsity comp vs. plain sparsity comp.

Cmd line args control the following:

color: use coloring prof: use profiler rev: use reverse mode check: check partials jax: use JAX sparsity comp sparse: use sparse partials fd: use finite difference partials show: show sparsity

openmdao.test_suite.scripts.jaxprof.main()