GPU ML Infra Intern: Speed Up Training & Profiling

PlusAI in Silicon Valley seeks a highly skilled systems/performance engineer to optimize BEV model training pipelines and GPU kernels. You will profile, implement custom kernels with CUDA or Triton, and push performance through LLM-assisted code generation and profiling workflows.

Key focus areas include memory management, PyTorch custom ops, and GPU utilization, with opportunities to influence autonomous vehicle perception workloads and high-performance computing practices.