feat: simple utility to get timestep count#34

Merged

EdanToledo merged 3 commits intomainfrom feat/total_timestep_util

Aug 29, 2024

+24

Commits on Jul 23, 2024

feat: timestep util from discussions in #33 .
callumtilbury
committed

Commits on Jul 24, 2024

fix: use lax.cond for jax control flow.
callumtilbury
committed

Commits on Jul 26, 2024

Merge branch 'main' into feat/total_timestep_util
callumtilbury
authored