feat: adopt a more JAX-like API
#211Comparing
jax-api-changes
(de77436
) withmain
(1977814
)+1%
- Untouched
- 36
Unknown Walltime execution environment detected
Using the Walltime instrument on standard Hosted Runners will lead to inconsistent data. For the most accurate results, we recommend using CodSpeed Macro Runners: bare-metal machines fine-tuned for performance measurement consistency.Benchmarks
36 total
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py+17%
110 µs94.2 µs
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py+16%
1,020.9 µs881.8 µs
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py+8%
126.7 µs117.7 µs
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py+7%
1.7 s1.6 s
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py+4%
136.3 ms130.9 ms
test_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py+4%
671.7 ms646.1 ms
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py+3%
13.6 ms13.1 ms
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py+3%
96.5 µs93.8 µs
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py+3%
388.9 µs378.8 µs
test_benchmark_moffat_conv[run]
tests/jax/test_benchmarks.py+2%
194.9 ms190.1 ms
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py+2%
214.8 µs211.3 µs
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py+1%
560.5 µs554.4 µs
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py+1%
162.9 ms161.1 ms
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py+1%
4.5 ms4.4 ms
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py+1%
48.6 µs48.2 µs
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py+1%
374.8 µs372.7 µs
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py+1%
1.1 ms1.1 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py+1%
11.5 ms11.5 ms
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py0%
43.8 ms43.6 ms
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py0%
6.9 µs6.8 µs
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py0%
436.5 µs435.6 µs
test_benchmark_moffat_conv[run]
tests/jax/test_benchmarks.py0%
2.3 s2.3 s
test_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py0%
5.4 s5.4 s
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py0%
435.2 µs435.2 µs
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py0%
34.9 s34.9 s
Commits
Click on a commit to change the comparison rangeBase
main
1977814
+61.35%
fix: do not need these changes
a54105d
5 days ago
by beckermr-1.29%
fix: make sure to convert to jax array
8f8d097
5 days ago
by beckermr-3.37%
fix: use `.at` syntax for interp images
e397991
5 days ago
by beckermr+4.19%
fix: more api fies
c152147
5 days ago
by beckermr+2.2%
tests: get tests to pass with .at syntax
fdbef71
5 days ago
by beckermr+0.63%
fix: more inplace ops
b1bd202
5 days ago
by beckermr-62.51%
Merge branch 'main' into jax-api-changes
de77436
3 days ago
by beckermr