Staff AI Infrastructure Engineer

Your Role in Our Mission

As a Staff AI Infrastructure Engineer, you will play a pivotal role in leading and managing infrastructure projects from training to production, mentoring team members, and influencing the strategic direction of our AI and hardware engineering initiatives.

Depending on interest and skills, responsibilities could include:

• Collaborating closely with software and research engineers to optimize and productionize training, experimentation, and enterprise product deployment infrastructure

• Designing, implementing, and maintaining individual microservices that collectively form complex, scalable backend systems

• Implementing tools, libraries and frameworks to speed up and enable new research and productization

• Be a part of planning and performing rapid prototyping of machine learning techniques applied to real-world scientific and enterprise semiconductor design and engineering problems

• Make improvements to model architectures, training, simulation, and compilation procedures

• Practice sustainable incident response and blameless postmortems

• Staying up-to-date with the latest advancements in AI, ML, and infrastructure technologies

• Mentoring and guiding junior colleagues, nurturing a collaborative, growth-oriented environment that promotes knowledge sharing and professional development

Qualifications:

• Bachelor’s degree or higher in Computer Science, Engineering, or a related field  

• 6+ years of experience in infrastructure engineering, with a focus on machine learning, distributed systems, and cloud computing

• Experience writing highly performant services and strong knowledge of Golang and/or Python.

• Specific expertise with managing and evolving Kubernetes in production and any cloud platform like GCP, AWS, Azure

• Expertise in monitoring & alerting, scalable testing, automation, CI/CD frameworks and best practices, infrastructure-as-code (Terraform, CloudFormation), and configuration management tools (Ansible, Puppet, Chef)

• Leadership and collaboration qualities, enthusiasm for real-world, responsible impact

• Excellent problem-solving and “getting things done” skills, and a proven ability to troubleshoot and optimize complex systems

• Strong written and verbal communication skills, with the ability to explain complex concepts to both technical and non-technical stakeholders across research and product

In addition, the following would be a significant advantage:

• Extensive experience with various database technologies, both relational (like PostgreSQL) and NoSQL (such as MongoDB, Cassandra, or DynamoDB). Proficiency in database design, sharding, replication, and tuning for high-robustness environments is highly desirable

• Applied experience with machine learning, preferably modern deep learning architectures (e.g. Transformers, CNNs, vision-language models, deep reinforcement learning)

• Experience using TensorFlow, PyTorch, Jax, NumPy, Pandas or similar ML/scientific libraries

• Comfort with probabilistic programming languages (e.g. Tensorflow Probability)

Equal Employment Opportunity Statement

Normal Computing is an Equal Opportunity Employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, veteran status, or any other legally protected status.

Accessibility Accommodations

Normal Computing is committed to providing reasonable accommodations to individuals with disabilities. If you need assistance or an accommodation due to a disability, please let us know at [email protected].

Privacy Notice

By submitting your application, you agree that Normal Computing may collect, use, and store your personal information for employment-related purposes in accordance with our Privacy Policy.