Axiom Math
Member of Technical Staff
Palo Alto +4 · On-site · Staff · Full-time · Posted Apr 3, 2026
- Location
- Palo Alto
US Remote
London
UK Remote
Portland
About the job
Leads research projects applying machine learning and AI to mathematical discovery and reasoning, from designing experiments and building solutions to publishing and communicating results.
What you'll do
- Set research agendas, run large-scale experiments, and publish research on mathematical structures.
- Collaborate with mathematicians to turn open problems into benchmarkable ML objectives and develop reproducible solutions.
- Communicate research findings and mentor junior researchers.
What they ask for
- Advanced Python coding and experience with modern ML frameworks.
- Deep expertise in large-scale training, reinforcement learning, program synthesis, and neuro-symbolic techniques.
- Strong foundation in higher mathematics and formal proof systems.
- Research experience in experiment design, rigorous analysis, and peer-reviewed publications.
Tools
Python, PyTorch, JAX, TensorFlow, Lean, Coq, Isabelle
AI focus
R&D in machine learning and AI for mathematical discovery and reasoning, including deep learning, neuro-symbolic methods, automated theorem proving, and mathematical research.
Model-written summary · the original posting is authoritative