Previous Job
Previous
CUDA Engineer
Ref No.: 20-05832
Location: Menlo Park, California
Pay Rate : $ 108.00 /Hour
Role & Responsibilities

* Support GPU kernel optimizations, unit testing, benchmarking, and PyTorch integration. Looking for candidates with AI experience.


Example tasks


* 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


Skill sets we're looking for

* Must: CUDA kernel optimization
* 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