# Computational Scientist

**Company:** [Voltai](https://hotfix.jobs/companies/voltai)
**Location:** Palo Alto, CA, California
**Role:** ML Engineering
**Skills:** C++, CUDA, Mpi, Fem, Fvm, Bem, Krylov Methods, Algebraic Multigrid, Neural Operators, Fno, Deeponet, Hypre, Petsc, Trilinos, Nccl
**Posted:** 2026-02-19

> Develops and scales MPI+CUDA PDE solvers and neural operators for electromagnetic simulations on GPU clusters for IC design. Requires PhD in computational physics/math, expertise in numerical methods, C++/CUDA, HPC, and neural operators for physics problems.

## Job Description

## What You'll Work On
- Develop and scale MPI+CUDA PDE solvers for electrostatics, charge transport, and electromagnetic field problems on complex 3D IC geometries across multi-node GPU clusters
- Tune and extend AMG preconditioners, Krylov solvers, and mesh pipelines for performance and correctness at scale
- Build and train neural operators (FNO, DeepONet, GNO, and variants) as high-fidelity surrogates for PDE-based field solvers
- Design simulation pipelines that generate training data for neural operator models — including sampling strategies, mesh handling, and physical consistency checks
- Validate everything: analytical solutions, published benchmarks, and cross-validation between field solvers and learned surrogates

## Required
- PhD in computational physics, applied mathematics, computational engineering, or a closely related field
- Deep expertise in numerical PDE methods: FEM, FVM, or BEM — weak formulations, quadrature, convergence, error analysis
- Strong C++ and CUDA — writing and optimizing kernels, memory hierarchy, multi-GPU programming
- Multi-node HPC: MPI, domain decomposition, collective communication, strong/weak scaling
- Sparse linear algebra at depth: Krylov methods, algebraic multigrid, preconditioning strategies
- Hands-on experience with neural operators (FNO, DeepONet or equivalent) — training, architecture design, and evaluation on PDE datasets
- Solid understanding of AI for Science methodology: how to design datasets from simulations, handle out-of-distribution generalization, and ensure physical consistency of learned models

## Strongly Preferred
- Experience with HYPRE, PETSc, and Trilinos
- Familiarity with multi-node GPU clusters: NCCL, CUDA-aware MPI, NVLink topologies
- Published work in neural operators, physics-informed ML, or scientific HPC
- IC design domain knowledge: device physics, semiconductor materials, layout data formats

## Similar roles

- [Software Engineer, AI Agents](https://hotfix.jobs/jobs/94686d85-8544-4cc4-a8cf-41545c3d9d95) - Cloudflare
- [CoDesign & NextGen Performance Engineer](https://hotfix.jobs/jobs/2cb728b7-a9ea-44b0-b16a-47364021f7be) - Cerebras Systems - Sunnyvale, CA
- [Research Engineer, Privacy](https://hotfix.jobs/jobs/3220c5d0-0e5f-4869-b133-1ed126b4208a) - OpenAI - San Francisco, CA - $380k – $445k/yr
- [Research Engineer](https://hotfix.jobs/jobs/b4d18018-e64d-4828-a8b4-303225328e17) - Console - San Francisco, CA - $200k – $350k/yr
- [Software Engineer, AI Platform](https://hotfix.jobs/jobs/7ddad000-3e1c-47cb-a45b-31d8e72abc4f) - Notion - San Francisco, CA - $180k – $201k/yr

**Apply:** https://hotfix.jobs/jobs/33428295-01a4-43ea-a102-1389f06d1cb3
**Canonical:** https://hotfix.jobs/jobs/33428295-01a4-43ea-a102-1389f06d1cb3