[WIP] feat: adopt a more JAX-like API
#211Comparing
jax-api-changes
(1469f2f
) withmain
(5f1e186
)+48%
- Improvement
- 1
- Untouched
- 35
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.Different runtime environments detected
Some benchmarks with significant performance changes were compared across different runtime environments, which may affect the accuracy of the results. Learn more about regression causes.Benchmarks
36 total
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py+48%
1,302.2 µs881.1 µs
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py+19%
146.1 µs122.4 µs
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py+8%
55 µs50.8 µs
test_benchmark_moffat_conv[run]
tests/jax/test_benchmarks.py+5%
193.7 ms184.9 ms
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py+5%
1.7 s1.6 s
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py+4%
167.8 ms161.3 ms
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py+4%
591.3 µs569 µs
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py+4%
135.6 ms130.9 ms
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py+1%
13 ms13 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py+1%
11.6 ms11.6 ms
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py0%
104.9 µs104.6 µs
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py0%
114.7 µs114.4 µs
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py0%
436.9 µs436 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py0%
48.9 µs48.9 µs
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py0%
375.1 µs374.7 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py0%
434.8 µs434.5 µs
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py0%
104.9 µs104.9 µ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_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py0%
2 s2 s
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py0%
34.9 s34.9 s
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py0%
1.1 ms1.1 ms
test_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py0%
656.1 ms656.5 ms
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py0%
1.4 ms1.4 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py0%
121.8 ms122.1 ms
Commits
Click on a commit to change the comparison rangeBase
main
5f1e186
+61.35%
fix: do not need these changes
a54105d
28 days ago
by beckermr-1.29%
fix: make sure to convert to jax array
8f8d097
28 days ago
by beckermr-3.37%
fix: use `.at` syntax for interp images
e397991
28 days ago
by beckermr+4.19%
fix: more api fies
c152147
28 days ago
by beckermr+2.2%
tests: get tests to pass with .at syntax
fdbef71
28 days ago
by beckermr+0.63%
fix: more inplace ops
b1bd202
27 days ago
by beckermr-62.51%
Merge branch 'main' into jax-api-changes
de77436
25 days ago
by beckermr+46.6%
Merge branch 'main' into jax-api-changes
1469f2f
13 days ago
by beckermr