OneTrust's mission is to enable organizations to use data and AI responsibly. The company is looking to drive the AI transformation across the organization by building reusable, scalable AI features and infrastructure, empowering every business unit to leverage AI.
Requirements
- 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.
- Solid foundation in Python programming.
- Deep interest in state-of-the-art AI.
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).
- Monitor, evaluate, and improve AI model performance in production environments.
- Troubleshoot and resolve issues related to data pipelines, model inference, and deployment.
- Conduct peer reviews and provide constructive feedback on machine learning code and processes.
Other
- An enthusiastic and innovative engineer with a strong desire to solve complex problems using artificial intelligence.
- Thrive in a collaborative, cross-functional environment and are excited to contribute to an ambitious mission of company-wide AI adoption.
- Excellent collaboration and communication abilities.
- Demonstrated curiosity and passion for AI, with a willingness to continuously learn and grow in the field.
- Office first culture, encouraging three days a week in office for most roles.