Siddharth Mishra-Sharma

Feature request: correlation of vector quantities

4

It is currently possible to use TreeCorr to compute the correlation functions of a scalar quantity (using kappa as proxy) as well as the shear, defined through g1 and g2....

Normalizing flow example

2

Implements a normalizing flow generative model, specifically the basic [Real NVP](https://arxiv.org/abs/1605.08803), for density estimation and sampling.

Based on a talk by Dan Levy (Mila) at the [Cosmic Connections](https://indico.flatironinstitute.org/event/3624/timetable/?view=standard) conference. Sketch of code: ```py from einops import repeat, rearrange import jax.numpy as jnp n_channels = 5 X_i...

Add ERWIN model implementation

2

## Warning Untested, generated by Claude Code. ## Summary - Implements the ERWIN (Efficient Hierarchical Transformer) model based on [the paper](https://arxiv.org/abs/2502.17019) - Port of the [original PyTorch implementation](https://github.com/maxxxzdn/erwin) to JAX/Flax...