Computational Scientist

The role

The person will develop and scale MPI+CUDA PDE solvers for electrostatics, charge transport, and electromagnetic field problems. They will build and train neural operators as high-fidelity surrogates for PDE-based field solvers.

What you will do

• 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 as high-fidelity surrogates for PDE-based field solvers
• Design simulation pipelines that generate training data for neural operator models
• Validate analytical solutions, published benchmarks, and cross-validation between field solvers and learned surrogates

Required qualifications

• 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

Nice to have

• 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

About the company

Voltai is developing world models, and agents to learn, evaluate, plan, experiment, and interact with the physical world. They are starting out with understanding and building hardware; electronics systems and semiconductors where AI can design and create beyond human cognitive limits.