search

CDCgov / DynODE

CDC/CFA/Predict/Scenarios ODE Modeling Framework
Apache License 2.0
3 stars 0 forks source link

optimize `external_i`, `seasonality`, and `vaccination_rate` functions with `jax.jit` #245

Open arik-shurygin opened 2 months ago

arik-shurygin commented 2 months ago

reduce their reliance on external parameters, try to completely isolate the functions to only take inputs and not rely on self, test this out to see if it reduces the runtime of fitting.

This should be done in conjunction with #231 as a part of performance improvement push

SamuelBrand1 commented 2 months ago

Hows this going?

arik-shurygin commented 3 weeks ago

Hows this going?

being explored here: https://github.com/CDCgov/DynODE/tree/gpu-optim