Principal Software Engineer - Artificial Intelligence / Machine Learning

Key Skills

Aartificial intelligence, AWS, Azure, Effective communication skills, Google Cloud Platform (GCP), HuggingFace, Machine learning techniques, PyTorch, TensorFlow

Job Description

Your Mission

As a Machine Learning Engineer, you will play a pivotal role in researching, developing, and implementing state-of-the-art AI models and workflows in Python. You will focus on creating production-ready, reusable components and robust pipelines that can be leveraged across multiple business domains. Your work will be at the heart of the company’s AI platform, enabling teams to seamlessly adopt and scale AI capabilities for a wide range of use cases.

Your Responsibilities

Development:

• Design, build, and optimize machine learning models, pipelines, and APIs using Python and state-of-the-art AI frameworks.
• Develop reusable, production-grade components that facilitate rapid AI solution deployment across business units.
• Implement configuration-driven workflows (e.g., YAML) to support scalable and flexible AI system orchestration.
• Ensure high-quality code through comprehensive unit, integration, and exploratory testing (‚â•80% coverage or as per team standards).

Collaboration:

• Partner with backend, frontend, and product teams to integrate AI models into scalable, user-facing features.
• Participate in technical planning, architecture reviews, and sprint ceremonies to align AI development with platform and business goals.
• Share knowledge and collaborate with team members to foster a culture of continuous learning and innovation.

Support:

• Monitor, evaluate, and improve AI model performance in production environments.
• Troubleshoot and resolve issues related to data pipelines, model inference, and deployment.
• Document workflows, best practices, and model configurations to ensure reproducibility and transparency.

Code Review:

• Conduct peer reviews and provide constructive feedback on machine learning code and processes.
• Advocate for best practices in ML engineering, testing, and automation.

Continuous Learning:

• Stay current with advancements in AI/ML, actively exploring and experimenting with new techniques, architectures, and frameworks.
• Bring a spirit of curiosity, experimentation, and passion for AI-even if you have not been able to apply it fully in previous roles.

You Are

An enthusiastic and innovative engineer with a strong desire to solve complex problems using artificial intelligence. You have a solid foundation in Python programming and a deep interest in state-of-the-art AI, whether or not your current role has allowed you to fully pursue it. You thrive in a collaborative, cross-functional environment and are excited to contribute to an ambitious mission of company-wide AI adoption.

Your Experience Includes 

• Bachelor's or Master's degree in Computer Science, Engineering, or a related field.
• 10+ years of professional software engineering experience, with a strong focus on frontend development.
• Hands-on experience with one or more modern AI/ML frameworks (such as TensorFlow, PyTorch, Hugging Face, scikit-learn, or similar).
• Experience designing and deploying production-grade machine learning models and pipelines.
• Familiarity with configuration-driven workflows and orchestration using tools like YAML.
• Experience with cloud-based ML platforms (Azure, GCP, AWS) and containerization (Docker) is a plus.
• Strong testing, debugging, and documentation skills.
• Excellent collaboration and communication abilities.
• Demonstrated curiosity and passion for AI, with a willingness to continuously learn and grow in the field.

For California, Colorado, Connecticut, Nevada, New York, Rhode Island, and Washington-based candidates: the annual base pay range for this role is listed below. Within this range, individual pay is determined by several factors, including location, job-related skills, work experience, and relevant education and/or training. This role may also be eligible for discretionary bonuses, equity, and/or commissions, as well as benefits.

Salary Range

$139,725 - $209,587.50 USD

Where we Work

We are embracing an office first culture, encouraging three days a week in office for most roles, with meaningful opportunities to collaborate and celebrate in person.

Each role may have specific requirements or flexibility depending on the scope of the position, so we encourage you to verify this with your recruiter during your first interview.

Benefits

As an employee at OneTrust, you will be part of the OneTeam. That means you’ll receive support physically, mentally, and emotionally so that you can do your best work both in and out of the office. This includes comprehensive healthcare coverage, flexible PTO, equity RSUs, annual performance bonus opportunities, retirement account support, 14+ weeks of paid parental leave, career development opportunities, company-paid privacy certification exam fees, and much more. Specific benefits differ by country. For more information, talk to your recruiter or visit onetrust.com/careers.

Resources 

Check out the following to learn more about OneTrust and its people:

• OneTrust Careers on YouTube
• @LifeatOneTrust on Instagram

Your Data

You have the right to have your personal data updated or removed. You also have the right to have a copy of the information OneTrust holds about you. Further details about these rights are available on the website in our Privacy Overview. You can change your mind at any time and have your personal data removed from our database. In order to do this you must contact us and let us know you wish to be removed. The request should be made on the Data Subject Request Form.

Recruitment fraud warning: OneTrust is aware of scams involving false offers of employment with our company. The fraudulent jobs, interviews and job offers use fake websites, email addresses, group chat and text messages. Be aware that we never ask candidates for personal information, IDs or bank information during the interview process. We do not interview prospective candidates via instant message or group chat, and do not require candidates to purchase products or services, or process payments on our behalf as a condition of any employment offer. Please note that any legitimate interview availability requests will come directly from a OneTrust recruiter with an "@onetrust.com" email address. You may also receive legitimate emails from "@us.greenhouse-mail.io". Recruiters will only reach out to candidates who have applied for a role through our ATS (Greenhouse) or prospects via LinkedIn InMail. Job offers will come from a recruiter and may have a "@docusign.net" email address. For more information or if you have been targeted please reach out to askrecruiting@onetrust.com.