Repository navigation

Flax: add shape argument to `set_timesteps` - #690

Merged

pcuenca merged 2 commits into

mainfrom

fix-flax-schedulers

Oct 3, 2022

Merged

Flax: add shape argument to `set_timesteps`#690
pcuenca merged 2 commits into
mainfrom
fix-flax-schedulers

Commits on Sep 30, 2022
- Flax: add shape argument to set_timesteps
  pcuenca
  committed
  58d03eb
- style
  pcuenca
  committed
  f576cc6