Avatar for the GalSim-developers user
GalSim-developers
JAX-GalSim
BlogDocsChangelog

fix: ensure bounds vmaps properly and behaves correctly within JIT

#249Closed
Comparing
bounds-fix-again
(
0774b6a
) with
main
(
db83109
)
CodSpeed Performance Gauge
+39%
Improvement
2
Untouched
34

Benchmarks

36 total
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+44%
71.4 µs49.6 µs
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+34%
141.6 µs105.3 µs
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+28%
1,246.2 µs970.5 µs
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+21%
54.4 µs44.9 µs
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+20%
216.5 µs179.7 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+17%
52.7 µs44.9 µs
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-16%
11.6 ms13.8 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-14%
122.3 ms142 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+14%
123.1 µs108.2 µs
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+12%
482.5 µs429.3 µs
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+11%
130.6 µs117.1 µs
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+11%
2.7 ms2.4 ms
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-11%
1.5 s1.7 s
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-8%
3.8 ms4.1 ms
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+7%
118.5 µs110.5 µs
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-6%
154.7 ms165 ms
test_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-6%
289.5 ms307 ms
test_benchmark_moffat_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-5%
183.6 ms194.1 ms
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-5%
18.4 ms19.3 ms
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-5%
571 µs598.5 µs
test_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-3%
4.1 s4.3 s
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-3%
2 s2.1 s
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-3%
100.5 µs103.8 µs
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+3%
6.8 µs6.6 µs
test_benchmark_moffat_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-3%
2.3 s2.4 s

Commits

Click on a commit to change the comparison range
Base
typing-inits-cleanup
db83109
+43.65%
fix: only apply ~ to array bool
956f778
24 days ago
by beckermr
-43.59%
fix: try dynamic children
03c7869
23 days ago
by beckermr
-19.89%
fix: cast in a different way
97b8823
23 days ago
by beckermr
+69.63%
fix: finish dynamic bounds impl
997dadf
23 days ago
by beckermr
-10.69%
fix: be sure to test everything
0774b6a
23 days ago
by beckermr
© 2026 CodSpeed Technology
Home Terms Privacy Docs