View Our Website View All Jobs

CUDA Engineer

  • Support GPU kernel optimizations
  • Unit testing
  • Benchmarking
  • PyTorch integration

Skill sets we’re looking for

  • Must: CUDA kernel optimization (at least 3 years of CUDA coding experience) 
  • Requirement (but if really strong at CUDA, we can consider without): experience with PyTorch
  • Good to have: experience in open source development environment
  • Good to have: deep learning experience with various models

In specific the candidate will be focusing on:

  • Kernel authoring: write CUDA kernels for training clusters
  • Better engineering
    • Make our CUDA kernels (FBGEMM-GPU) to have better tests
    • Code refactoring for better readability, modularity, and extensibility
  • Methodical benchmarking to identify bottlenecks

Minimum of BSc in CS/CE/EE or equivalent 


Read More

Apply for this position

Apply with Indeed
We've received your resume. Click here to update it.
Attach resume as .pdf, .doc, .docx, .odt, .txt, or .rtf (limit 5MB) or Paste resume

Paste your resume here or Attach resume file