AI Optimization Engineer

Company Description

Join a multibillion-dollar global company that brings together amazing technology, people, and operational scale to become a powerhouse in the memory industry. Headquartered in Rancho Cordova, California, Solidigm combines elements of an established, successful technology company with the spirit, agility, and entrepreneurial mindset of a start-up. In addition to the U.S. headquarters and other facilities in the U.S., the company has international presence in Asia, Europe, and the Americas. Solidigm will continue to lead the world in innovating new Memory technologies with aspirations to be the #1 NAND memory company in the world. At Solidigm, we view problems as opportunities to define innovative solutions that hold the power to change the world and unleash the potential technological needs that the future holds. At Solidigm, we are One Team that fosters a diverse, equitable, and inclusive culture that embraces individual uniqueness and empowers us to bring our best selves to deliver excellence in support of Solidigm's vision and mission to be the go-to partner for optimized data storage solutions. You can be part of the takeoff of an innovative business that develops cutting-edge products, delivers strong business value for customers, provides an engaging workplace for its employees, and serves a greater impact on the world. This is a golden opportunity for the right applicant to join us and help design, build, and lead Solidigm. We want a diverse team of dedicated professionals who will not just be Solidigm team members but contribute to how we shape the future of the organization. We are seeking applicants who will grow and thrive in our culture; be customer inspired, trusting, innovative, team-oriented, inclusive, results driven, collaborative, passionate, and flexible.

Job Description

AI Optimization Engineer builds and validates machine learning workflows and infrastructure necessary to productize AI models and sustain them in production

In this job, you will Build and Validate machine learning workflows and infrastructure necessary to productize AI models and sustain them in production. You will work on preparing data for ML models at scale, building appropriate interfaces for ML model consumption, enabling ML Ops for continuous delivery and automation of ML pipelines.

Qualifications

• Bachelor in computer science or similar field with 3-5 years of experience or Master's with 2-3 year of experience.
• Hands-on development with Python.
• Experience with AI and machine learning frameworks (e.g., TensorFlow, PyTorch, ONNX)
• Practical knowledge of concepts in the following areas: debugging, coding techniques, object-oriented programming, drivers, remote debugging,
• Knowledge of software validation concepts: unit / functional/ integration / exploratory / performance / scalability tests,
• Methodical problem-solving skills,
• Fluency in English.

Preferred Qualifications:

• Experience with optimizing ML inference runtimes
• Experience with ML model training and optimization
• Experience with Docker and Kubernetes technologies