Infrastructure engineer at Causal Labs (AI startup) building and optimizing large distributed ML training and inference clusters for a Large Physics foundation Model, starting with weather prediction. Core stack: FSDP, DeepSpeed, NVIDIA GPUs, Python/C++, Kubernetes, Docker, GCP/AWS/Azure; on-site in San Francisco 5 days a week.
Sign in to see your match