Skip to content

Keep DG state as a PyTree and shard it over spatial cells - #39

Open
rogeriojorge wants to merge 4 commits into
agent/dg-autodifffrom
agent/dg-pytree-state
Open

Keep DG state as a PyTree and shard it over spatial cells#39
rogeriojorge wants to merge 4 commits into
agent/dg-autodifffrom
agent/dg-pytree-state

Merge branch 'agent/dg-autodiff' into agent/dg-pytree-state

f5e9981
Select commit
Loading
Failed to load commit list.

There are no checks for this commit