Software Engineer - Machine Learning Acceleration

We're hiring on behalf of a Haystack partner! The Role • Develop and maintain high-performance runtime libraries and drivers for machine learning applications and AI accelerators. • Design, develop, and deploy core components of a machine learning acceleration software stack. • Optimize AI workloads across custom hardware platforms by providing deep insights into performance bottlenecks. • Improve the performance of machine learning kernels and frameworks. • Manage the full development life cycle of runtime software, ensuring scalability, reliability, and usability. • Collaborate with cross-functional teams to ensure compiler outputs are optimized for custom hardware. What You'll Need • 3+ years of professional software development experience. • 2+ years of experience in designing or architecting new and existing systems, focusing on reliability and scaling. • Proficiency in at least one programming language. • Experience with the full software development life cycle, including coding standards, code reviews, source control, build processes, testing, and operations. • Experience in architecting, building, and operating distributed systems with a focus on high availability and fault tolerance. • Hands-on experience with cloud services in production environments. What's On Offer • Competitive annual salary ($165,200 - $223,600 USD) plus potential for sign-on payments and restricted stock units. • Comprehensive benefits package including health insurance, 401(k) matching, paid time off, and parental leave. • Opportunities for mentorship and career growth within a supportive team environment. • Flexibility in working hours to promote work-life balance. Apply via Haystack today!