PlasmaControl / DESC

Stellarator Equilibrium and Optimization Suite
MIT License
97 stars 26 forks source link

Export compiled objectives for common equilibrium resolutions? #1402

Open dpanici opened 4 hours ago

dpanici commented 4 hours ago

https://jax.readthedocs.io/en/latest/export/export.html

YigitElma commented 3 hours ago

In my last meeting with Egemen, we talked about this after I showed him some profiling. The long initialization and long compile times made him ask this. I think #1374 and general performance improvements on factorize_inear_constraints will make actual minimization start way earlier and further tricks like storing compiled code will be redundant. But might be worth trying.

Some concerns tho,

I feel like this is not too easy but anyway.