Software Engineer, Generalist, AI Inference

Job Description

As a member of the Dojo ML team, you will build AI frameworks and infrastructure for efficient inference on edge devices. You'll work on deploying neural networks on heterogeneous compute, collaborating with AI scientists and compiler engineers to optimize models, and implementing custom GPU kernels for efficient training and post-processing.

Requirements

• Degree in Engineering, Computer Science, or equivalent experience
• Proficiency in Python and C++
• Experience with PyTorch or another ML framework
• Familiarity with computer systems and architecture
• Experience with CUDA.

Responsibilities

• Build AI frameworks for edge devices
• Train and fine-tune networks on GPU clusters
• Deploy neural networks on various hardware
• Collaborate to compress models
• Design and implement custom GPU kernels.

Benefits

• Benefits include Aetna PPO and HSA plans with $0 payroll deduction, family-building benefits, dental and vision plans with $0 contribution, flexible spending accounts, 401(k) with employer match, company-paid life insurance, sick and vacation time, and employee discounts.