CHANGELOG.md
CITATION.cff
CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
docs/api.md
docs/guide.md
docs/index.md
docs/quickstart.md
examples/00_standalone.py
examples/01_quickstart.py
examples/02_reverse_grad.py
examples/03_engine_routing.py
examples/04_reverse_through_fused.py
examples/05_bring_your_own_ode.py
examples/06_stiff.py
examples/07_saveat_timeseries_fit.py
examples/08_fused_kernel_from_jax.py
examples/09_gpu.py
examples/README.md
examples/__init__.py
examples/assets/readme_gradient.png
examples/notebooks/getting_started.ipynb
gradsolve/__init__.py
gradsolve/api.py
gradsolve/base.py
gradsolve/dispatch.py
gradsolve/py.typed
gradsolve.egg-info/PKG-INFO
gradsolve.egg-info/SOURCES.txt
gradsolve.egg-info/dependency_links.txt
gradsolve.egg-info/requires.txt
gradsolve.egg-info/top_level.txt
gradsolve/cuda/__init__.py
gradsolve/cuda/_build.py
gradsolve/cuda/_codegen.py
gradsolve/cuda/_ffi_bridge.py
gradsolve/cuda/_fields.py
gradsolve/cuda/_stiff_fields.py
gradsolve/cuda/cuda_rosenbrock23.py
gradsolve/cuda/cuda_tsit5.py
gradsolve/cuda/kernels/rosenbrock23_template.cu.in
gradsolve/cuda/kernels/tsit5_template.cu.in
gradsolve/solvers/__init__.py
gradsolve/solvers/adaptive_imex.py
gradsolve/solvers/dense.py
gradsolve/solvers/diffrax_fallback.py
gradsolve/solvers/fixed_step_explicit.py
gradsolve/solvers/fixed_step_imex.py
gradsolve/solvers/fixed_step_tsit5.py
gradsolve/solvers/imex.py
gradsolve/solvers/method.py
gradsolve/solvers/record_jax.py
gradsolve/solvers/rodas5p_replay.py
gradsolve/solvers/rodas5p_step.py
gradsolve/solvers/rosenbrock23_step.py
gradsolve/solvers/tsit5_replay.py
gradsolve/solvers/tsit5_step.py
gradsolve/solvers/vern7_replay.py
gradsolve/solvers/vern7_step.py
gradsolve/warp/__init__.py
gradsolve/warp/_warp_kernel.py
gradsolve/warp/_warp_rosenbrock.py
gradsolve/warp/fused_backward.py
gradsolve/warp/fused_rosenbrock.py
gradsolve/warp/jax_field.py
gradsolve/warp/warp_ode.py
gradsolve/warp/warp_replay.py
gradsolve/warp/warp_rosenbrock.py
tests/__init__.py
tests/test_adjoint_replay.py
tests/test_api.py
tests/test_cuda_ffi_cpu.py
tests/test_dispatch.py
tests/test_f32_replay.py
tests/test_fjax_shape_contract.py
tests/test_grad_closure_y0.py
tests/test_jax_field.py
tests/test_ma_regression.py
tests/test_method_contract.py
tests/test_no_examples_import.py
tests/test_problem_protocol.py
tests/test_record_jax.py
tests/test_rodas5p_dense.py
tests/test_rodas5p_dense_saveat.py
tests/test_rodas5p_engine.py
tests/test_rodas5p_replay.py
tests/test_rodas5p_step.py
tests/test_saveat.py
tests/test_solvers_adaptive_imex.py
tests/test_solvers_fixed_step.py
tests/test_standalone_import.py
tests/test_steps.py
tests/test_tsit5_error_weights.py
tests/test_tsit5_replay.py
tests/test_vern7_engine.py
tests/test_vern7_replay.py
tests/test_vern7_step.py
tests/test_version_identity.py
tests/test_warp_import_guard.py
tests/_oracles/__init__.py
tests/_oracles/gradsolve_g2_spike_oracle.py
tests/_oracles/tier0_fused_adaptive_oracle.py