Staff Machine Learning Engineer, AI Generation Engine

The role

This person will own the end-to-end ML lifecycle from data exploration to scalable production deployment. They will design and build AI-first products using Large Quantitative Models and agentic frameworks.

What you will do

• Design, construct, and manage robust data pipelines for training, validation, and retraining of LQMs and agentic frameworks.
• Develop, implement, and test novel ML models and algorithms.
• Clean, transform, and engineer features from complex and large-scale datasets.
• Conduct deep analysis of model behavior, performance, and failure modes.
• Tune hyper-parameters and optimize model architecture.
• Collaborate with AI researchers, product managers, and SWEs to translate business objectives into ML roadmaps.
• Enforce engineering standards for code quality, system efficiency, and security.
• Drive technical execution with high autonomy.

Required qualifications

• BS in Software Engineering, Computer Science, or equivalent field of study
• 8+ years of postgraduate experience in software development
• Experience developing highly-available, performant, scalable ML systems, including large-scale data processing pipelines
• Strong expertise in Python (including the ML stack: PyTorch, TensorFlow, JAX, NumPy, Pandas)
• Long, successful history of driving the full ML lifecycle: from initial data exploration and hypothesis testing to architecture, model training, evaluation, and production deployment
• Deep proficiency in MLOps and software best practices, including CI/CD for ML, experiment tracking (e.g., Weights & Biases, MLflow), automated testing, and version control for both code and datasets

Nice to have

• MS or PhD in Software Engineering, Computer Science or equivalent experience
• Financial simulation or technical experience, risk simulation
• Equivalent experience includes tech leadership in a complex space, driving technical design and execution cross-collaboratively across multiple teams and organizations
• Experience with scalable software development on cloud computing platforms (e.g., GCP, AWS)

Benefits

• Comprehensive medical, dental, and vision coverage for employees and dependents with generous employer premium contributions
• Retirement savings with company matching
• Paid parental leave
• Inclusive family-building benefits
• Flexible paid time off
• Company-wide seasonal breaks
• Support for flexible work arrangements
• Opportunities for continuous learning and growth through on-the-job development, cross-functional collaboration, and access to internal learning and development programs

About the company

SandboxAQ is a high-growth company delivering AI solutions that address some of the world's greatest challenges. The company emerged from Alphabet Inc. as an independent, growth capital-backed company in 2022.