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

fix: ensure bounds vmaps as best as possible, more equinox errors, types for `__eq__`, `_BoundsD/I`, and `_PositionD/I`

#250
Comparing
bounds-fix-again-all-static
(
d5d20dd
) with
main
(
e2b1029
)
CodSpeed Performance Gauge
+45%
Improvement
12
Regression
1
Untouched
23

Benchmarks

36 total
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
×2.1
80.3 µs38.5 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+78%
56.9 µs32 µs
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+75%
56.3 µs32.1 µs
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+57%
125.1 µs79.7 µs
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+53%
135.8 µs88.8 µs
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+53%
502.5 µs329.2 µs
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+51%
217 µs143.8 µs
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+49%
135.7 µs91.3 µs
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+48%
7.6 µs5.1 µs
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+39%
119.3 µs85.9 µs
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+36%
7.3 µs5.3 µs
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+33%
14 ms10.5 ms
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-35%
523 µs803.1 µs
test_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+24%
311.6 ms251.6 ms
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+22%
166 ms136.4 ms
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+21%
3.9 ms3.3 ms
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+20%
1.8 s1.5 s
test_benchmark_moffat_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+18%
198.6 ms167.9 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+17%
974.4 µs832.1 µs
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+16%
1,016.2 µs876.4 µs
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+5%
19 ms18.2 ms
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+2%
573.4 µs564.5 µs
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+1%
333.5 µs331.7 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
438.2 µs437.6 µs
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
105.8 µs105.7 µs

Commits

Click on a commit to change the comparison range
Base
main
e2b1029
+0.03%
fix: be sure to test everything
0774b6a
5 days ago
by beckermr
+39.71%
fix: ensure we handle branches on bounds eq properly
93eb30f
5 days ago
by beckermr
-43.79%
fix: put back variable pytree def
004ada8
4 days ago
by beckermr
+1.31%
fix: make sure to send fits headers to galsim
7f97f11
4 days ago
by beckermr
-35.91%
Merge branch 'typing-inits-cleanup' into bounds-fix-again-all-static
a0d0ba6
4 days ago
by beckermr
+70.16%
Merge branch 'typing-inits-cleanup' into bounds-fix-again-all-static
996a5fb
4 days ago
by beckermr
-83.14%
Merge branch 'main' into bounds-fix-again-all-static
2143616
4 days ago
by beckermr
×2
Apply suggestion from @beckermr
63fe4b3
4 days ago
by beckermr
-51.6%
fix: allow python bool for static bounds eq
d0aab7e
4 days ago
by beckermr
+13.06%
test: update to latest submodule
de2bb72
4 days ago
by beckermr
+31.87%
doc: add docs
29df9c3
4 days ago
by beckermr
+9.71%
doc: be a bit more specific
2d53ab0
4 days ago
by beckermr
-88.93%
fix: dead code
1c94cf3
4 days ago
by beckermr
+47.03%
test: add test of bool eq api
0d07661
4 days ago
by beckermr
-15.15%
test: add tests of bounds and vmap
dea6915
4 days ago
by beckermr
-48.51%
test: add tests of bounds and vmap
1671b37
4 days ago
by beckermr
×2.2
doc: clarify
401e540
3 days ago
by beckermr
-58.76%
test: more tests for bnds int and vmap
0a1ddd1
3 days ago
by beckermr
-5.87%
test: add tests for type conversion
aa8d587
3 days ago
by beckermr
+1.43%
doc: add comment
a7efe2d
3 days ago
by beckermr
+40.56%
fix: rename for clarity
eb2748a
3 days ago
by beckermr
-41.68%
test: add tests of includes and simpler bounds init
215b412
2 days ago
by beckermr
-0.44%
test: run faster?
863a99e
2 days ago
by beckermr
+2.12%
test: use more splits
137edd1
2 days ago
by beckermr
-0.41%
test: less logging
c7348ef
2 days ago
by beckermr
-1.88%
Apply suggestion from @beckermr
642b4ae
2 days ago
by beckermr
-34.67%
fix: do not cat this giant file
afe868f
2 days ago
by beckermr
+37.71%
test: update to the latest test suite
6eff057
20 hours ago
by beckermr
-2.78%
test: one more submod update
87b1ecf
19 hours ago
by beckermr
+9.02%
test: update submod again
5dabac4
19 hours ago
by beckermr
+36.33%
fix: add position private classes
d5d20dd
18 hours ago
by beckermr
© 2026 CodSpeed Technology
Home Terms Privacy Docs