Principal Software Engineer - Artificial Intelligence / Machine Learning jobs in United States
cer-icon
Apply on Employer Site
company-logo

OneTrust · 1 day ago

Principal Software Engineer - Artificial Intelligence / Machine Learning

OneTrust is a leader in enabling organizations to use data and AI responsibly. The role of Principal Software Engineer focuses on researching, developing, and implementing advanced AI models and workflows, collaborating with cross-functional teams to drive AI transformation across the organization.

Artificial Intelligence (AI)ComplianceCyber SecurityData GovernanceInformation TechnologySoftware
check
H1B Sponsor Likelynote

Responsibilities

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)
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
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
Conduct peer reviews and provide constructive feedback on machine learning code and processes
Advocate for best practices in ML engineering, testing, and automation
Stay current with advancements in AI/ML, actively exploring and experimenting with new techniques, architectures, and frameworks

Qualification

PythonMachine LearningAI FrameworksProduction-grade ModelsCloud-based ML PlatformsConfiguration-driven WorkflowsDebugging SkillsCollaboration SkillsDocumentation SkillsContinuous Learning

Required

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

Benefits

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

Company

OneTrust

company-logo
OneTrust is the AI-Ready Governance platform helping organizations accelerate innovation while ensuring responsible data use.

H1B Sponsorship

OneTrust has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2023 (8)
2022 (104)
2021 (110)
2020 (94)

Funding

Current Stage
Late Stage
Total Funding
$2.07B
Key Investors
Generation Investment ManagementSoftBank Vision FundTCV
2023-07-24Series Unknown· $150M
2023-03-31Secondary Market· $1B
2021-04-06Series C· $210M

Leadership Team

leader-logo
Kabir Barday
Founder, CEO
linkedin
leader-logo
Guido Torrini
Chief Financial Officer
linkedin
Company data provided by crunchbase