SandboxAQ · 1 month ago
Senior Research Engineer, CFD Algorithm Development
United States
Full-time
Remote
Mid, Senior Level
$154K/yr - $216K/yr
3+ years expSandboxAQ is a high-growth company delivering AI solutions that address some of the world's greatest challenges. The role involves leading the development of a differentiable Computational Fluid Dynamics (CFD) framework and optimizing chemical reactor designs using advanced computational techniques.
Artificial Intelligence (AI)Cyber SecurityInformation TechnologyQuantum ComputingSaaS
Responsibilities
Lead development of differentiable CFD solvers for non-isothermal and reactive flow simulations on mulit-node multi-GPU hardware using JAX
Design, implement, and containerize code to ensure reproducibility, scalability, and ease of deployment across HPC GPU environments
Develop optimized parallel linear solvers using FFT or other matrix decompositions
Extend existing implementation of immersed boundary method to complex geometries and non-periodic boundary conditions
Extend and validate CFD capabilities from single-node to multi-node GPU systems for both non-reactive and reactive flow regimes
Integrate transport equations for up to 50 chemical species using linear numerical methods
Validate simulation accuracy against benchmark results and fixed-bed reactor configurations
Develop gradient-based optimization scripts to iteratively modify reactor designs based on objective functions defined by catalyst activity and selectivity
Collaborate closely with AI model developers to link catalyst property predictions with reactor-scale flow optimization
Generate comprehensive technical reports summarizing algorithms, scalability, differentiability, and validation results for milestone reviews
Qualification
Required
PhD or MS in Computation Physics, Mechanical or Chemical Engineering, Computer Science or equivalent discipline
3+ years (including PhD) of hands-on experience in CFD code development, preferably on GPUs
Proven expertise building scalable scientific software on multi-GPU systems using JAX, PyTorch, CUDA, MPI frameworks or similar
Full fluency across all of: numerical linear algebra, numerical methods for solving PDEs, sparse and dense linear solvers, immersed boundary methods
Excellent Python programming skills, with experience in JAX a prerequisite
Ability to work independently, manage multiple deliverables, and produce clear technical documentation
Preferred
PhD + 3 years industry experience in Computational Physics, Mechanical/Chemical Engineering, Computer Science or related field
Proven expertise in CFD code development for non-isothermal flow, transport equations, and reactive flow modeling
Proven expertise with differentiable computational fluid dynamics, and/or gradient-based optimization of complex CFD objectives (i.e. adjoint optimization, auto-diff a.s.o)
Proven experience across the full stack of scientific software development including PR submission, code review, unit-testing, integration testing, containerization with Docker and enroot,, and deployment on GPU-HPC systems using k8s, Google Cloud Batch, slurm or similar
Proven expertise in chemical reactor and multi-physics modeling (e.g., reactive flow, energy and mass transport in fixed-bed or catalytic systems)
Demonstrated track record of delivering project milestones and publishing high-quality technical results
Benefits
Stock options depending on employment type
Generous learning opportunities
Medical/dental/vision
Family planning/fertility
PTO (summer and winter breaks)
Financial wellness resources
401(k) plans
And more
Company
SandboxAQ
SandboxAQ develops AI and quantum technology solutions that enhance biopharma, cybersecurity, and materials science.
51-200 employees
H1B Sponsorship
SandboxAQ has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (7)
2024 (6)
2023 (3)
2022 (5)
2021 (1)
2020 (5)
Funding
Current Stage
Late StageTotal Funding
$975MKey Investors
Eric Schmidt Angel InvestmentsMichael J. Fox Foundation
2025-04-04Series E· $150M
2024-12-18Series E· $300M
2024-10-29Grant· $25M
Leadership Team
Jack Hidary
Chief Executive Officer
Joerg Rathenberg
Head Of Marketing
Recent News
2025-12-30
Fast Company Middle East | The future of tech, business and innovation.
2025-12-24
Company data provided by crunchbase