Solutions Architect - AI / ML - Training & GPU infra

Tasks

• Debug ML workloads in customer environments
• Design distributed training architectures
• Design large scale inference architectures
• Evaluate networking and interconnect trade offs
• Operate multi node GPU workloads
• Optimize and scale multi node training
• Tune GPU performance
• Use GPU profiling and monitoring

Perks/Benefits

• Remote work

Skills/Tech-stack

DeepSpeed | Distributed Systems | GPU Monitoring | GPU Profiling | Infiniband | JAX | Kubernetes | NVIDIA NVLink | PyTorch | PyTorch distributed | Slurm