Systems Engineer V

• Company Apex Systems
• No. of Openings 10+
• Salary Not Disclosed
• Work Type on-site

Job Description :

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 GPUs
• 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 were looking for
   

Must:

• GPU optimization or experience with PyTorch
• 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