Skip to content
VoltaiVoltaiPalo Alto, CA

Computational Scientist

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.

Salary not listed
On-siteML Engineering

About the role

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

Skills

C++CUDAMpiFemFvmBemKrylov MethodsAlgebraic MultigridNeural OperatorsFnoDeeponetHyprePetscTrilinosNccl

Similar roles

ML Engineering jobs
Cloudflare

Software Engineer, AI Agents

CloudflareUnited States

Build and ship production AI agents on Cloudflare's edge platform using Workers, Durable Objects, and AI tools. Requires strong TypeScript/Rust experience, observability expertise, and hands-on LLM tooling for evals, safety, and multi-agent systems.

Salary not listed
On-siteML Engineering
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k/yr
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k/yr
Hybrid5+ YOEML Engineering