Skip to content

Commit 0dd0bf0

Browse files
prishajain1prishajain1
committed
added jax.lax.scan and spatial sharding
1 parent ad74cb3 commit 0dd0bf0

1 file changed

Lines changed: 426 additions & 309 deletions

File tree

0 commit comments

Comments
 (0)