Staff Machine Learning Engineer, AI Generation Engine
SandboxAQ · Remote · United States ·
COMPENSATION
$196k – $294k
Listed on the employer's job board (Ashby)
- Staff level role
- Fully remote
- Python + PyTorch
- MLOps focus
- Equity included
Join the AI Generation Engine (SAIGE) team at SandboxAQ as a founding engineer. You will own the end-to-end ML lifecycle, from data exploration to production deployment, focusing on Large Quantitative Models and agentic frameworks. This position requires rapid prototyping and translating cutting-edge AI concepts into functional MVPs.
What you'll do
- Design and manage data pipelines for training and retraining LQMs
- Develop and test novel ML models aligned with product objectives
- Engineer features from complex datasets to optimize model accuracy
- Analyze model behavior and tune hyper-parameters for production efficiency
- Collaborate with researchers and engineers to define ML roadmaps
- Enforce engineering standards for code quality and security
- Drive technical execution and make independent design decisions
What you'll need
- BS in Software Engineering, Computer Science, or equivalent field
- 8+ years of postgraduate experience in software development
- Experience building scalable, high-performance ML systems
- Strong expertise in Python, PyTorch, TensorFlow, JAX, NumPy, Pandas
- Proven track record managing the full ML lifecycle to production
- Proficiency in MLOps, CI/CD, experiment tracking, and version control
Nice to have
- MS or PhD in Software Engineering or Computer Science
- Experience in financial simulation or risk simulation
- Tech leadership experience in complex, cross-functional spaces
- Experience with scalable development on GCP or AWS
Stack
- Python
- PyTorch
- TensorFlow
- JAX
- NumPy
- Pandas
- Weights & Biases
- MLflow
- GCP
- AWS
What you get
- Competitive base salary and performance-based bonuses
- Equity participation in the company
- Comprehensive medical, dental, and vision coverage
- Retirement savings plan with company matching
- Paid parental leave and family-building benefits
- Flexible paid time off and seasonal breaks
- Access to internal learning and development programs