Axiom builds the formal foundations and tooling that let machines and mathematicians verify proofs and reason about logic — including AXLE, our interactive proof environment. Backed by Madrona and Menlo.
About the role
Lead core discovery projects (e.g., successors to PatternBoost ): set research agendas, design and run large-scale experiments to reveal latent mathematical structures, and publish high-impact papers in top AI and mathematics venues. Collaborate with research mathematicians to identify open problems to tackle, formulate them into benchmarkable ML objectives, build reproducible pipelines, and iterate toward state-of-the-art solutions.
What they're looking for
- Advanced coding in Python and modern ML frameworks (PyTorch, JAX, TensorFlow)
- Deep expertise in large-scale training, reinforcement learning, program synthesis, and neuro-symbolic techniques
- Strong foundation in higher mathematics (algebra, analysis, combinatorics) and formal proof systems (Lean, Coq, Isabelle)
- Demonstrated research acumen: experiment design, rigorous analysis, and a track record of peer-reviewed publications
- Exceptional scientific writing and presentation abilities, tailoring complex ideas to diverse audiences
- Proven collaborator who thrives in interdisciplinary teams with mathematicians, engineers, and product stakeholders
More about this role
Machine Learning & AI for Mathematical Discovery and Reasoning — R&D at the intersection of deep learning, neuro-symbolic methods, automated theorem proving, and pure/applied mathematics.
Lead core discovery projects (e.g., successors to PatternBoost ): set research agendas, design and run large-scale experiments to reveal latent mathematical structures, and publish high-impact papers in top AI and mathematics venues.
Collaborate with research mathematicians to identify open problems to tackle, formulate them into benchmarkable ML objectives, build reproducible pipelines, and iterate toward state-of-the-art solutions.
Communicate results broadly through peer-reviewed publications, conference talks, open-source releases, and internal briefings that translate research insights into business value.
Mentor and coach junior researchers by providing technical guidance, rigorous code reviews, and career development support, fostering a culture of excellence and collaboration.
Advanced coding in Python and modern ML frameworks (PyTorch, JAX, TensorFlow).
Deep expertise in large-scale training, reinforcement learning, program synthesis, and neuro-symbolic techniques.
Strong foundation in...
Browse similar: AI jobs · AI startup jobs · Startup jobs · San Francisco Bay Area