System Modeling (Computation)

About Unconventional

Since 2022, AI has entered the mainstream, reshaping entire industries from education and software development to fundamental consumer behaviors. This revolution has created an unprecedented demand for computation - a demand that is now fundamentally limited by energy, not just in the datacenter, but at a global scale.

At Unconventional, our mission is to solve this. We are rethinking computing from the ground up to build a new foundation for AI that is 1000x more efficient. We're doing this by exploiting the rich physics of semiconductors, mapping neural networks directly to the device physics rather than relying on layers of inefficient abstraction.

The Role

As a Member of Technical Staff, System Modeling (Computation), you will be part of a hands-on R&D team building simulation frameworks that enable evaluation and rapid iteration across all layers of unconventional physics-based computing systems for machine learning workloads. “Extreme co-design” is our guiding principle.

System Modeling is a multi-disciplinary effort, and the team we’re building reflects that. The role involves development of physics-based system models, GPU-accelerated ML system simulations, and cross-layer system integration. You don’t need to be an expert in all of these, but you have to be very strong in at least one, and solid in the rest.

Responsibilities

You will be responsible for one or more of the following tasks:

• Architecting Foundational Solvers: Building large-scale, GPU-accelerated, high-fidelity numerical differential equation solvers (ODE, SDE, CDE, PDE). You will build tools that enable rapid iteration, multiple architectures, and rich metrics/visualization, leveraging frameworks best suited for scientific ML (e.g., JAX, PyTorch, or custom CUDA/Triton kernels).
• Bridging Physics and Machine Learning: Developing physics-based surrogate models of device- and system-level behavior in unconventional compute. You will create clean, composable abstractions that expose algorithm–hardware tradeoffs and enable cross-layer optimization via end-to-end autodiff.
• Extreme Co-Design & Collaboration: Working closely with hardware and algorithm teams to understand their simulation needs, supporting everything from high-level algorithm development to the low-level verification of novel, analog hardware.

Minimum Qualifications

• Education
• Scientific Machine Learning (SciML) & Numerical Methods (replaces Dynamical computation knowledge)

• Deep expertise in numerical differential equation solvers (e.g., ODE, SDE, DDE) and their implementation on parallel architectures (e.g., Rosenbrock methods, Euler-Maruyama, adjoint methods, implicit solvers).
• Experience with high-performance, customized GPU kernel development for numerical methods, including GPU memory optimization and multi-GPU scaling.
• Experience building effective neural network surrogate models (e.g., Neural ODEs) for complex dynamical systems.

• ML and systems fluency

Preferred Qualifications

We are looking for well-rounded candidates. While the minimum qualifications focus on the core modeling and ML expertise, candidates who possess the following qualities will be the most impactful.

• Software Engineering & API Craftsmanship
• Dynamic Systems Knowledge
• Systems Thinking

Why Join Us?

• The Mission: Redefine computing for the next 50 years by solving the fundamental energy limitation of AI at a global scale.
• The Impact: Shape the company's future as a foundational team member. Enjoy massive ownership and an outsized opportunity to drive change.
• The Challenge: Dive into deeply complex, intellectually stimulating, and unsolved problems at the cutting edge of multiple, converging fields.
• The Perks: A comprehensive package including best-in-class health benefits, 401k matching, truly unlimited PTO, and complimentary meals when working from our Palo Alto office.