Machine Learning Engineer — Physics AI

The role focuses on benchmarking and researching compression technologies for large-scale scientific and simulation datasets, integrating these methods into PyTorch and distributed training pipelines, and producing technical reports and publications. It involves running extensive experiments on physics AI workloads, analyzing performance impacts, and collaborating with research partners.