7 projects
sbibm-jax
Simulators and datasets for simulation-based inference benchmarks in JAX/NumPyro
heal-swin-nnx
HealSwin: spherical HEALPix Swin V2 U-Net in JAX/Flax NNX, with RoPE and seam-exact shifted windows
gensbi
GenSBI is a library for Simulation-Based Inference using generative methods in JAX.
gensbi-examples
Examples for the GenSBI library
review-mode-mcp
MCP server for Review Mode — manage review annotations from any AI agent
gPCS
Code for the paper Deepening gamma-ray point-source catalogues with sub-threshold information
ConcreteDropout
Concrete Dropout implementation for Tensorflow 2 and PyTorch
