AI Research Scientist - Computational Science
About us:
Axiomatic AI is building a new class of AI systems designed to reason with the rigor of the scientific method. By combining deep learning with formal logic and physics-based modeling, we create verifiable, interpretable AI systems that collaborate with and support human researchers in high-stakes scientific and engineering workflows.
Our mission, 30×30, is to deliver a 30× improvement in the speed, accessibility, and cost of semiconductor and photonic hardware development by 2030.
We aim to revolutionize hardware design and simulation in these industries and are building a team of highly motivated professionals to bring these innovations from research into commercial products.
Position overview:
As an AI Research Scientist specialized in computational science, you will lead the development of next-generation agentic-AI-aware computational science tools for physics simulations and engineering analysis and optimization. Your responsibilities will include designing, prototyping/evaluating, and deploying advanced scientific computing systems that integrate AI reasoning with traditional numerical methods. You will architect solutions for electromagnetic simulations and other computational physics tools while ensuring they maintain scientific rigor and verifiable accuracy. Close collaboration with our cross-functional team, consisting of Mathematicians, Physicists, AI Engineers, and Software Engineers will be essential to deliver tools that transform how engineers and scientists approach computational problems.
Your mission:
- AI-Driven Scientific Computing: Lead the development of tools and AI systems for the automation of scientific computing workflows, focused on electromagnetic simulations (FDTD, FEM) and associated inference, uncertainty quantification, and optimization tasks.
- Research & Innovation: Drive cutting-edge research at the intersection of AI and computational science. Implement agent-aware tooling for common computational science workflows alongside systems for verification of numerical accuracy and scientific rigor.
- Technical Leadership: Mentor junior scientists and engineers, establish best practices for scientific computing tool development, and guide architectural decisions that balance performance, accuracy, and usability.
- Collaboration & Integration: Work closely with domain experts in photonics and other fields to understand requirements and deliver tools that address real engineering challenges.
- Documentation & Outreach: Produce detailed technical documentation, publish research findings in top scientific computing and AI venues, and present results to both technical and business stakeholders.
Key requirements:
- PhD (or 3+ years of relevant industry experience) in Mathematics, Computer Science, Engineering, Physics, or a related field with focus on computational engineering/physics/statistics, control/optimization theory and practice, scientific computing, or numerical analysis.
- Background in PDE solving (or Bayesian) methods for inverse problems & uncertainty quantification with solid baseline understanding of numerical simulation & optimization methods, computational statistics, and scientific computing principles.
- Strong track record in developing numerical methods, frameworks for computational science, or tools for scientific computing.
- Expert-level proficiency in Python and/or other relevant scientific computing ecosystems (JAX, Julia, MPI, ...).
- Strong communication skills with ability to explain technical concepts to diverse audiences.
- Self-motivated with a proactive, solution-oriented mindset and ability to lead technical initiatives.
Nice to have:
- Electromagnetic Simulations: Hands-on experience with FDTD, FEM, or other methods for solving Maxwell's equations; bonus for experience working with dedicated tools like MEEP, FEMWELL, Tidy3D, Lumerical, COMSOL, or other.
- PDE Simulations: Experience with CFD, structural mechanics, heat transfer, or other multi-physics simulation domains; knowledge of FEniCS, OpenFOAM, or similar frameworks.
- Applied Reinforcement Learning: Practical experience applying RL to online decision-making problems; bonus for application to (tool-using) AI systems.
- Agentic AI Systems: Enthusiasm for building autonomous agents, multi-agent systems, or AI systems that interact with external tools and APIs.
- Proven excellence through publications in top-tier venues (SIAM, IEEE Transactions, ACM, AISTATS, NeurIPS, ICLR, ICML, JMLR, etc.)
- Contributions to open-source scientific computing or machine learning projects.
- Track record of deploying production machine learning systems or computational tools.
- Experience with cloud computing platforms and containerization (Docker, Kubernetes).
What we offer:
- Competitive compensation
- Stock Options Plan: Empowering you to share in our success and growth.
- Cutting-Edge Tools: Access to state-of-the-art tools and collaborative opportunities with leading experts in artificial intelligence, physics, hardware and electronic design automation.
- Work-Life Balance: Flexible work arrangements in one of our offices with potential options for remote work.
- Professional Growth: Opportunities to attend industry conferences, present research findings, and engage with the global AI research community.
- Impact-Driven Culture: Join a passionate team focused on solving some of the most challenging problems at the intersection of AI and hardware.
Why join us?
At Axiomatic_AI, you will be working on technology that drives innovation in AI for scientific and engineering applications in line with our 30 x 30 mission. This is your opportunity to contribute to the development of new AI architectures that can reason coherently and produce interpretable and verifiable solutions. Consequently, see those ideas commercialized into products that will shape the future of hardware and computing, while collaborating with a global team of engineers and AI specialists. We believe in pushing the boundaries of what is possible and continuously seek to redefine the intersection of AI, with focus on formal consistency. If you're ready to take your expertise in artificial intelligence and physics to the next level, we want to hear from you!
Worried about not meeting every qualification? Studies show that women and people of color are less likely to apply for jobs unless they meet every listed requirement. At Axiomatic-AI, we are dedicated to creating a diverse, inclusive, and authentic workplace. If this role excites you but your background doesn’t perfectly match every qualification, we still encourage you to apply. You could be the perfect fit for this position or another opportunity with us.
Apply for this job
*
indicates a required field

