ShardTensor: Domain Parallelism for Scientific Machine Learning

Corey Adams, Peter Harrington, Akshay Subramaniam +4

Scientific Machine Learning (SciML) faces unique challenges for extreme-resolution data, with mitigations that often fail to scale or degrade the accuracy of trained models. While…