jax
7e05f94c - fix pmap nesting bug, sketch out index splitting

Commit

7 years ago

fix pmap nesting bug, sketch out index splitting

References

#272 - initial spmd groundwork

Author

mattjj

mattjj

Parents

Loading