Software Engineer, Performance & Benchmarking

Weights & Biases Weights & Biases · Data AI · Bellevue, WA +1 · Technology

Software Engineer focused on performance and benchmarking for AI infrastructure, likely involving GPU optimization and system-level analysis. The role is not directly building AI models but supporting the infrastructure that runs them.

What you'd actually do

  1. Drive performance and benchmarking efforts for CoreWeave's AI infrastructure.
  2. Develop and maintain performance testing frameworks and tools.
  3. Analyze and identify performance bottlenecks across the stack, from hardware to software.
  4. Collaborate with engineering teams to implement performance improvements.
  5. Report on performance metrics and provide insights to guide optimization efforts.

Skills

Required

  • Strong understanding of computer systems architecture, including CPUs and GPUs.
  • Experience with performance analysis and benchmarking tools.
  • Proficiency in at least one systems programming language (e.g., C++, Rust, Go).
  • Familiarity with cloud computing environments.
  • Excellent problem-solving and analytical skills.

Nice to have

  • Experience with AI/ML workloads and frameworks (e.g., TensorFlow, PyTorch).
  • Knowledge of distributed systems.
  • Familiarity with containerization technologies (e.g., Docker, Kubernetes).

What the JD emphasized

  • AI infrastructure
  • performance
  • benchmarking
  • GPU performance