Nathaniel Starkman

release v0.2.1

1

For JAX v0.7.0 compatibility. Thanks!

Tests: benchmarks

2

~Followup to #67.~ Requires: 1. Codspeed to be set up 2. the tag `run-benchmarks` to be added.

I found 2 reproducible tracer leaks in #67 1. `jax.numpy.linalg.inv` 2. `jax.numpy.linalg.solve` These have `pytest.param(..., marks=pytest.xfail)` so they are verifiably a bug and included in the CI. They are also...

https://ui.adsabs.harvard.edu/abs/2024A%26A...682A..31B/abstract

✨ feat(dynamics): class-based relaxation time calculators

1

This PR changes from the verbose ```python class MethodFlag: ... def relax_time_specific_method(): ... @dispatch def relaxation_time(type[MethodFlag], ...): return relax_time_specific_method() ``` to the simpler ```python class Method: ... def __call__(*args): ......