jax
da724d31 - single-axis-var version runs! and some cleanup

Commit

6 years ago

single-axis-var version runs! and some cleanup

References

#272 - initial spmd groundwork

Author

mattjj

mattjj

Committer

mattjj

mattjj

Parents

Loading