SandboxAQ · 1 day ago
Senior Research Engineer, CFD Algorithm Development
United States
Full-time
Remote
Mid, Senior Level
$151K/yr - $239K/yr
3+ years expSandboxAQ is a high-growth company delivering AI solutions that address some of the world's greatest challenges. The Engineering Team is looking for a Senior Research Engineer, CFD Algorithm Development to develop and deploy a differentiable Computational Fluid Dynamics (CFD) framework on multi-GPU and multi-node systems to optimize chemical reactor designs.
Artificial Intelligence (AI)Cyber SecurityInformation TechnologyQuantum ComputingSaaS
Responsibilities
Lead the development of differentiable CFD solvers using JAX for non-isothermal and reactive flow simulations on multi-node, multi-GPU hardware
Optimize reactor scale flow by integrating transport equations for up to 50 chemical species and linking catalyst property predictions with gradient-based optimization
Deliver scalable scientific software that enables physics-based optimization of complex reactor geometries through immersed boundary methods and parallel linear solvers
Develop Differentiable Solvers: Lead the creation of differentiable CFD frameworks in JAX, extending capabilities to complex geometries and non-periodic boundary conditions using immersed boundary methods
Scalable GPU Engineering: Design, implement, and containerize code (Docker/enroot) to ensure reproducibility and scalability across multi-node GPU HPC environments
Numerical Optimization: Develop optimized parallel linear solvers (FFT or matrix decompositions) and gradient-based scripts to iteratively modify reactor designs based on catalyst activity
Validate & Benchmark: Rigorously validate simulation accuracy against benchmark results and fixed-bed reactor configurations for both non-reactive and reactive flow regimes
Cross-Functional Collaboration: Partner with AI model developers to integrate property predictions and generate comprehensive technical reports summarizing algorithm scalability and differentiability
Qualification
Required
PhD or MS in Computational Physics, Mechanical/Chemical Engineering, Computer Science, or equivalent
3+ years (including PhD) of hands-on experience in CFD code development, specifically on GPUs
Proven ability to build scalable software on multi-GPU systems using JAX, PyTorch, CUDA, or MPI frameworks
Full fluency in numerical linear algebra, PDE numerical methods, sparse/dense linear solvers, and immersed boundary methods
Excellent Python programming skills, with JAX proficiency as a prerequisite
Ability to manage multiple deliverables and produce clear technical documentation for milestone reviews
Preferred
PhD + 3 years of industry experience in a related computational field
Proven expertise in non-isothermal flow, transport equations, and chemical reactor multi-physics (e.g., mass transport in fixed-bed systems)
Expertise in adjoint optimization or auto-diff for gradient-based optimization of complex CFD objectives
Experience with the full stack of scientific development, including CI/CD, unit testing, and deployment via K8s, Slurm, or Google Cloud Batch
Demonstrated ability to deliver complex project milestones and publish high-quality technical results
Benefits
Comprehensive medical, dental, and vision coverage for employees and dependents with generous employer premium contributions
Retirement savings with company matching
Paid parental leave
Inclusive family-building benefits
Flexible paid time off
Company-wide seasonal breaks
Support for flexible work arrangements that enable sustainable performance
Opportunities for continuous learning and growth through on-the-job development, cross-functional collaboration, and access to internal learning and development programs
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