Braze is seeking a Principal AI Engineer to design, develop, and deploy AI solutions that address complex business challenges. You will apply advanced expertise in artificial intelligence, including machine learning and natural language processing, to implement these technologies in production-grade systems.

What You'll Do

Develop innovative, scalable AI solutions for real business problems.
Drive the full lifecycle of projects from conception to deployment, ensuring alignment with business objectives.
Own highly open-ended projects end-to-end, from the analysis of business requirements to the deployment of solutions.
Manage complex data sets, design efficient data processing pipelines, and work on robust models.
Work closely with other AI engineers, product managers, and stakeholders to ensure solutions meet business needs and enhance user satisfaction.
Write clear, concise, and comprehensive technical documentation for all projects and systems developed.
Stay updated on the latest developments in the field and explore and prototype new technologies and approaches.
Develop and maintain high-quality machine learning services, prioritizing robust engineering practices and user-centric development.

What We're Looking For

Master’s degree in Computer Science, Machine Learning, Statistics, Engineering, Mathematics, or a related field.
Deep understanding and practical experience in machine learning and natural language processing, especially LLMs.
Strong foundational knowledge in statistical modeling, probability, and linear algebra.
Extensive practical experience with curating datasets, training models, analyzing post-deployment data, and developing robust metrics to ensure model reliability.
Experience developing and maintaining machine learning services for real-world applications at scale.
Strong Python programming skills.
High standards for code craftsmanship, producing maintainable, testable, and production-ready code.
Proficiency with Docker.
Knowledge of system design and cloud infrastructure for secure and scalable AI solutions.
Proficiency with AWS.
Proven track record in driving AI projects with strong technical leadership.
Excellent communication skills when engaging with both technical and non-technical stakeholders.

Nice to Have

Experience with natural language processing for legal applications.
Proficiency with Terraform.
React and Node.js experience.

Technical Stack

Python
Docker
AWS
Terraform
React
Node.js

Team & Environment

You will work closely with other AI engineers, product managers, and stakeholders. Braze fosters a diverse, remote-first culture empowered with flexibility and resources, where innovation thrives, every contribution is valued, and the environment is high-paced.

Benefits & Compensation

Competitive compensation and benefits.

Work Mode

This is a remote-first position based in India.

G-P is a proud Equal Opportunity Employer, and we are committed to building and maintaining a diverse, equitable and inclusive culture that celebrates authenticity. We prohibit discrimination and harassment against employees or applicants on the basis of race, color, creed, religion, national origin, ancestry, citizenship status, age, sex or gender (including pregnancy, childbirth, and pregnancy-related conditions), gender identity or expression (including transgender status), sexual orientation, marital status, military service and veteran status, physical or mental disability, genetic information, or any other legally protected status.