Senior Machine Learning Engineer, AI Generation Engine

The role

This person will take ownership of the end-to-end ML lifecycle for the AI Generation Engine. 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.

Required qualifications

• BS in Software Engineering, Computer Science, or equivalent field of study
• 5+ 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
• Deep proficiency in MLOps and software best practices, including CI/CD for ML, experiment tracking, automated testing, and version control

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

• Competitive salary, equity and annual bonus
• 401k matching at 50% up to IRS maximum contribution
• Unlimited PTO plus a summer and winter break (one week each)
• Twelve weeks of fully paid parental leave in the US, with another 8 weeks for birthing parents
• $750 equipment, software, and office furniture budget
• $100 per month for wellness (physical or mental) and $100 for home office bills
• Top-notch medical, dental and vision insurance for you and your dependents with all premiums covered at 95% for employees
• Family Planning support (fertility, surrogacy, adoption) through Carrot

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.