jax
780106f8 - moving pxla flattening/chunking to api.py, wip

Commit

6 years ago

moving pxla flattening/chunking to api.py, wip

References

#272 - initial spmd groundwork

Author

mattjj

mattjj

Committer

mattjj

mattjj

Parents

Loading