Unknown Company

GPU ML Infra Intern: Speed Up Training & Profiling

santa clara, ca • Posted 1 weeks ago
Onsite Full Time Electrical & Energy Engineering

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.

#J-18808-Ljbffr
Back to Job Search