Skip to content

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.

Palo Alto, CACaliforniaML EngineeringOnsite

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

ML Engineer, Agentic Systems

ML Engineer building and improving agentic systems powered by LLMs for multimodal video understanding, reasoning, and creative editing tasks at an AI-native video platform. Requires strong production ML experience with transformers, fine-tuning, and experimental rigor.

175k – 275kNew York, NYML EngineeringOn-siteLLMsPython

Software Engineer, Agents

Design and build agentic systems for AI-native video creation, integrating LLMs and evaluation frameworks to power creative workflows. Requires 5+ years building ML/agentic systems in production.

175k – 275kNew York, NYML EngineeringOn-site5+ YOERAGLLMs

Research Scientist II

Research Scientist II building and improving fraud risk models and scam detection systems using audio, behavioral, and metadata signals. Requires an advanced degree and 3+ years of applied ML experience with Python and modern ML frameworks.

160k – 185kUnited StatesML EngineeringRemote3+ YOELLMsKeras

Research Engineer, Post-Training

Research engineer focused on post-training LLMs and agents for legal work. Requires hands-on experience training open-weight models and strong Python/research engineering skills.

231k – 340kSan Francisco, CAML EngineeringHybridSftRLHF

AI Engineer

Build full-stack AI prototypes and agentic systems to pressure-test venture ideas. Requires 3+ years building production AI applications with strong frontend/backend fluency and frontier coding agent expertise.

150k – 190kMountain View, CAML EngineeringOn-site3+ YOESQLAPIs