vmap index that uses hitran.line_strength

HajimeKawahara / exojax

🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt

http://secondearths.sakura.ne.jp/exojax/

MIT License

45 stars 14 forks source link

vmap index that uses hitran.line_strength #439

Closed HajimeKawahara closed 9 months ago

HajimeKawahara commented 9 months ago

Some correction for vmap(line_strength). I think we should put None as the Tref argument in vmapped line_strength Yui's code in #437 is now included in tests/integration/api/api_premodit_to_direct.py Looks fine but confirm it?

HajimeKawahara commented 9 months ago

something wrong... close