optimize external_i, seasonality, and vaccination_rate functions with jax.jit
#245
Labels
enhancement
New feature or request
Comments
arik-shurygin
changed the title
external_i, seasonality, and vaccination_rate functions for jax.JITexternal_i, seasonality, and vaccination_rate functions with jax.JIT
arik-shurygin
changed the title
external_i, seasonality, and vaccination_rate functions with jax.JITexternal_i, seasonality, and vaccination_rate functions with jax.jit
|
Hows this going? |
being explored here: https://github.com/CDCgov/DynODE/tree/gpu-optim |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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
The text was updated successfully, but these errors were encountered: