4 projects
cmb-augr
Fisher-matrix forecasting for CMB B-mode polarization experiments
jaxht
JAX-native spherical harmonic transforms (spin-0/2 on-grid + off-grid NUFFT): GPU-capable, differentiable, and minimal-dependency (pure JAX + numpy).
mlxmc
MCMC samplers in Apple MLX
mbody
M-body: a differentiable particle-mesh N-body toy in MLX on Apple Silicon, targeting autodiff dlnP(k)/df_NL