Software Engineer, Systems ML

Software Engineer, Systems ML position involves developing machine learning compiler frameworks and driving next generation hardware software codesign for AI domain specific problems.

Requirements

• Proven C/C++ programming skills
• Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
• Experience in AI framework development or accelerating deep learning models on hardware architectures
• Knowledge of GPU, CPU, or AI hardware accelerator architectures
• Experience working with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT
• Experience in accelerating libraries on AI hardware, similar to cuBLAS, cuDNN, CUTLASS, HIP, ROCm etc.
• Experience with compiler optimizations such as loop optimizations, vectorization, parallelization, hardware specific optimizations such as SIMD
• Demonstrated ability to integrate AI tools to optimize/redesign workflows and drive measurable impact
• Experience adhering to and implementing responsible, ethical AI practices
• Demonstrated ongoing AI skill development and staying current with emerging AI technologies

Benefits

• Generous Paid Time Off
• 401k Matching
• Retirement Plan
• Bonus
• Equity