[Remote] Senior Machine Learning Engineer, AI Generation Engine
Note: The job is a remote job and is open to candidates in USA. SandboxAQ is a high-growth company delivering AI solutions that address some of the world's greatest challenges. They are seeking a Senior Machine Learning Engineer to take ownership of the end-to-end ML lifecycle, from data exploration and model development to scalable production deployment, focusing on building AI-first products using Large Quantitative Models (LQMs).ResponsibilitiesDesign, construct, and manage robust data pipelines for the training, validation, and continuous retraining of Large Quantitative Models (LQMs) and agentic frameworksDevelop, implement, and rigorously test novel ML models and algorithms, defining appropriate metrics to ensure model performance aligns with high-level product objectivesContribute to the efforts in cleaning, transforming, and engineering features from complex and large-scale datasets to optimize LQM performance and predictive accuracyConduct deep analysis of model behavior, performance, and failure modes, tuning hyper-parameters and optimizing model architecture for efficiency, speed, and accuracy in a production contextCollaborate closely with AI researchers, product managers, and SWEs to translate high-level business objectives into actionable ML development and deployment roadmapsChampion and enforce exceptional engineering standards for code quality, system efficiency, and security in a prototyping environmentSkillsBS in Software Engineering, Computer Science, or equivalent field of study5+ years of postgraduate experience in software developmentExperience developing highly-available, performant, scalable ML systems, including large-scale data processing pipelinesStrong 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 deploymentDeep 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 datasetsMS or PhD in Software Engineering, Computer Science or equivalent experienceFinancial simulation or technical experience, risk simulationEquivalent experience includes tech leadership in a complex space, driving technical design and execution cross-collaboratively across multiple teams and organizationsExperience with scalable software development on cloud computing platforms (e.g., GCP, AWS)BenefitsCompetitive salary, equity and annual bonus401k matching at 50% up to IRS maximum contributionUnlimited 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 billsTop-notch medical, dental and vision insurance for you and your dependents with all premiums covered at 95% for employeesFamily Planning support (fertility, surrogacy, adoption) through CarrotCompany OverviewSandboxAQ develops AI and quantum technology solutions that enhance biopharma, cybersecurity, and materials science. It was founded in 2016, and is headquartered in Palo Alto, California, USA, with a workforce of 51-200 employees. Its website is https://www.sandboxaq.com.Company H1B SponsorshipSandboxAQ has a track record of offering H1B sponsorships, with 7 in 2025, 6 in 2024, 3 in 2023, 5 in 2022, 1 in 2021, 5 in 2020. Please note that this does not guarantee sponsorship for this specific role.