nuclear-multimessenger-astronomy / nmma

A pythonic library for probing nuclear physics and cosmology with multimessenger analysis
https://nuclear-multimessenger-astronomy.github.io/nmma/
GNU General Public License v3.0
30 stars 58 forks source link

Added flax interpolation for KN models #265

Closed ThibeauWouters closed 11 months ago

ThibeauWouters commented 11 months ago

This pull requests adds another interpolation type to the kilonova surrogate models allowing users to train neural networks with Flax.