Infrastructure engineer at Causal Labs, an AI startup building a Large Physics foundation model for causal prediction of physical systems. You design, scale, and debug distributed training systems across thousands of GPUs using PyTorch/JAX stacks (FSDP, DeepSpeed, Megatron) to make large-scale training fast and reliable.
Sign in to see your match