AbstractCosmologicalEmulators.jl
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
library, whose performance is excellent on the CPU for the kind of small neural networks (NN) that we employ. We plan to include other frameworks, such as Lux.jl
, in order to support models running on the GPU. If you want include a new NN/GP framework, feel free to open a PR or write a message.
Roadmap to v1.0.0
Step | Status | Comment |
---|---|---|
Interface with SimpleChains.jl |
:heavy_check_mark: | Implemented |
Support for vectorization | :white_check_mark: | Implemented, needs some polishing |
Interface with Lux.jl |
:construction: | WIP |
GPU support | :construction: | WIP |
Stable API | :construction: | WIP |
Authors
- Marco Bonici, PostoDoctoral Researcher at INAF-IASF Milano