AbstractCosmologicalEmulators.jl is the central Julia package within the the CosmologicalEmulators Github organization, which defines methods and structs used by the other packages hosted by the organization.
In this moment the emulators here used are based only on the SimpleChains.jl and Lux.jl libraries. If you want include a new NN/GP framework, feel free to open a PR and/or get in touch with us.
| Step | Status | Comment |
|---|---|---|
Interface with SimpleChains.jl |
✔️ | Implemented |
Interface with Lux.jl |
✔️ | Implemented |
| Support for vectorization | ✔️ | Implemented |
AD Rules ChainRules |
✔️ | Implemented |
| Robust emulators initialization | ✔️ | Implemented, needs some polishing |
| GPU support | ✔️ | Implemented, needs some polishing |
| Gaussian Process support | 🚧 | Work in progress |
| Stable API | 🚧 | Work in progress |
- Marco Bonici, PostDoctoral Researcher at Waterloo Centre for Astrophysics
- Marius Millea, Researcher at UC Davis and Berkeley Center for Cosmological Physics