8 projects
dorian-raytrace-mpi
Deflection Of Rays In Astrophysical Numerical simulations - Full-sky ray-traced weak gravitational lensing maps from cosmological simulations
jax-cadre
Constraint-Aware Descent Routine Executor — JAX-native constrained optimization
furax-cs
GPU Powered CMB Parametric Component Seperation using Furax and JAX
jax-hpc-profiler
A comprehensive benchmarking and profiling tool designed for JAX in HPC environments, offering automated instrumentation, strong/weak scaling analysis, and performance visualization.
jaxdecomp
JAX bindings for the cuDecomp library
jax-grid-search
Distributed grid search in JAX
jax-progress
Progress meters for JAX loops and scans
jaxpm
A simple Particle-Mesh implementation in JAX
