Ai Software Engineer

Summary

Join our team as an AI Software Engineer and be at the forefront of cutting-edge AI research and software engineering. You will transform state-of-the-art ML models into efficient, production-ready systems. Collaborate with research and engineering teams to implement novel AI architectures, ensuring high performance and reliability. This role demands strong software engineering fundamentals and ML expertise. You will shape the future of AI infrastructure by tackling challenging problems in machine learning. We offer a competitive salary, comprehensive benefits, and opportunities for professional development.

Requirements

• Bachelor’s or Master’s degree in Computer Science, Engineering, Mathematics, or a related field
• Minimum of 5 years of experience in software engineering with a focus on AI/ML
• Strong proficiency in C++ and Python for software development
• Experience binding C/C++ into Python (binding in other languages a bonus)
• Proficiency with C++ build systems like CMake
• Proven experience with PyTorch; familiarity with MLX is a plus
• Strong understanding of computer science fundamentals and algorithms
• Familiarity with AI/ML concepts, including LLM and computer vision model architectures
• Familiarity with distributed systems and parallel computing concepts

Responsibilities

• Build, document, optimize, and maintain (through CI/CD and testing) AI related software packages and services
• Design and implement efficient, production-ready AI/ML systems
• Collaborate closely with research, engineering, and product teams to understand requirements
• Integrate and optimize AI/ML algorithms and research derived by R&D into platform packages and services
• Debug complex issues within our AI stack

Preferred Qualifications

• Experience with CUDA programming and GPU driven compute acceleration
• Knowledge of ML model quantization and optimization techniques
• Familiarity with MLOps tooling and practices
• Contributions to open-source ML projects
• Experience developing software within distributed systems
• Experience contributing to AI frameworks

Benefits

• Competitive salary and performance-based incentives
• Comprehensive health, dental, and vision benefits package
• 401k Match
• $200/mos Health and Wellness Stipend
• $400/year Continuing Education Credit
• Free parking, for in-office employees
• Unlimited Approved PTO
• Parental, Bereavement Leave
• Supplemental Life Insurance