Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ Inc.

Job Overview

Location

Remote

Job Type

Full-time

Full Job Description

📋 Description

• SandboxAQ is at the forefront of addressing global challenges through advanced AI solutions, and our AI Generation Engine (SAIGE) team is pivotal in this mission. We are seeking a highly accomplished Staff Machine Learning Engineer to spearhead the end-to-end lifecycle of our AI-first SaaS products. This role is ideal for a hands-on engineer with a proven track record of taking machine learning models from initial concept and experimentation through to scalable, production-grade deployment. You will be instrumental in designing and rapidly building innovative products that leverage SandboxAQ's proprietary Large Quantitative Models (LQMs) and sophisticated agentic frameworks.
• As a founding engineer on the SAIGE team, you will act as a critical bridge between cutting-edge AI research and the development of functional, real-world Minimum Viable Products (MVPs). Your primary objective will be to rapidly iterate on potential solutions, build and evaluate new models, and drive tangible outcomes with a focus on speed and effectiveness. This position offers a unique opportunity to shape the future of AI product development within a dynamic, high-growth company.
• Your responsibilities will encompass the full spectrum of ML engineering, including but not limited to: Data Acquisition and Curation, Infrastructure development, Pre-Training strategies, rigorous Evaluations, and advanced Fine-Tuning techniques. You will be a key player in a diverse, collaborative team comprising software engineers, ML experts, product managers, and user experience researchers, contributing significantly to the efficient and effective enablement of the groundbreaking technologies developed at SandboxAQ.
• Key responsibilities include: Designing, constructing, and managing robust data pipelines essential for the training, validation, and continuous retraining of Large Quantitative Models (LQMs) and complex agentic frameworks. This involves ensuring data integrity, scalability, and efficiency throughout the pipeline.
• Developing, implementing, and rigorously testing novel ML models and algorithms. You will be responsible for defining appropriate performance metrics that align with high-level product objectives and ensuring that models meet stringent quality and efficacy standards.
• Leading the critical effort in cleaning, transforming, and engineering features from complex, large-scale datasets. This work is vital for optimizing LQM performance, enhancing predictive accuracy, and unlocking new capabilities.
• Conducting deep, analytical investigations into model behavior, performance characteristics, and potential failure modes. You will actively tune hyper-parameters and optimize model architecture to achieve superior efficiency, speed, and accuracy in production environments.
• Collaborating closely with AI researchers, product managers, and fellow software engineers to translate high-level business objectives into actionable ML development and deployment roadmaps. This requires strong communication and strategic thinking.
• Championing and enforcing exceptional engineering standards for code quality, system efficiency, and security, even within a fast-paced prototyping environment. Your commitment to best practices will set a high bar for the team.
• Driving technical execution with a high degree of autonomy, making critical design and implementation decisions independently. This role demands initiative, problem-solving skills, and a proactive approach to challenges.
• Contributing to the architectural design of our ML systems, ensuring they are scalable, maintainable, and aligned with long-term company goals. You will have a significant impact on the technical direction of the SAIGE team.
• Staying abreast of the latest advancements in machine learning, deep learning, and agentic AI, and proactively identifying opportunities to incorporate these innovations into our product development process.
• Participating in code reviews, knowledge sharing sessions, and mentoring junior engineers, fostering a culture of continuous learning and technical excellence within the team.
• Working with cloud-based infrastructure to deploy and manage ML models, ensuring reliability and performance at scale. This includes leveraging services from platforms like GCP or AWS.
• Developing and maintaining comprehensive documentation for ML models, pipelines, and systems, ensuring knowledge transfer and maintainability.
• Engaging with stakeholders to understand their needs and provide technical insights, ensuring that ML solutions effectively address business requirements.
• Contributing to the strategic planning of the SAIGE team, helping to define the technical roadmap and prioritize initiatives.
• This role is a unique opportunity to be at the forefront of AI innovation, working on challenging problems with a talented and passionate team, and making a tangible impact on the world.

🎯 Requirements

• BS in Software Engineering, Computer Science, or a related technical field, or equivalent practical experience.
• 8+ years of postgraduate experience in software development, with a significant portion focused on machine learning systems.
• Proven experience in developing and deploying highly-available, performant, and scalable ML systems, including experience with large-scale data processing pipelines.
• Deep expertise in Python and its associated ML stack (e.g., PyTorch, TensorFlow, JAX, NumPy, Pandas).
• Demonstrated success in managing the full ML lifecycle: from initial data exploration and hypothesis testing through architecture design, model training, evaluation, and production deployment.
• Strong proficiency in MLOps principles and software engineering best practices, including CI/CD for ML, experiment tracking (e.g., Weights & Biases, MLflow), automated testing, and version control for both code and datasets.

🏖️ Benefits

• Competitive base salary, performance-based incentives or bonuses, and equity participation.
• Comprehensive medical, dental, and vision coverage for employees and dependents with generous employer premium contributions.
• Retirement savings plan with company matching.
• Paid parental leave and inclusive family-building benefits.
• Flexible paid time off and company-wide seasonal breaks.
• Opportunities for continuous learning and growth through on-the-job development, cross-functional collaboration, and access to internal learning and development programs.

Skills & Technologies

Python

AWS

GCP

TensorFlow

PyTorch

Senior

Remote

Degree Required

Ready to Apply?

Apply Externally

You will be redirected to an external site to apply.

SandboxAQ Inc.

Visit Website

About SandboxAQ Inc.

SandboxAQ is an enterprise AI company spun out of Alphabet in 2022, focused on applying large quantitative models to solve complex problems in cybersecurity, encryption, sensing, and simulation. Its software and hardware solutions combine AI with quantum-inspired methods to help government and Fortune 500 clients secure data, accelerate materials discovery, and optimize sensing for healthcare and navigation. Based in Palo Alto with global offices, the company partners with large systems integrators and cloud vendors to deploy scalable, physics-aware AI platforms for defense, life sciences, and financial services organizations.

View Company Profile