Show more filters
Banner image for Kake

Mathematician with Python - AI Trainer - Remote

Kake

3.0
8 reviews
Kake
Job Type   /   Job Level
Contract   /   Senior Executive
Company Location
Estonia

We're building a talent pool of Mathematics professionals with Python proficiency to contribute to project-based AI development initiatives, focused on evaluating and enhancing frontier AI models.

Designed for mathematicians who enjoy deep technical problem-solving, this pipeline role is for those looking to apply their analytical expertise to evaluate and push the boundaries of frontier AI models, relying on domain-specific tools such as Z3, cvc5, SageMath, or Macaulay2, with verifiable, code-graded answers run inside isolated Linux environments.


Key Responsibilities

  • Identify an appropriate mathematical software package and build problems whose solution genuinely hinges on that tool's core capabilities.
  • Develop full Python solutions for each problem, providing input files where applicable.
  • Establish the correct answer and define the acceptable margin for the AI model's response, based on the mathematical context.
  • Run the problem against the AI model across multiple parallel attempts, analyzing where it succeeds or falls short, and adjusting difficulty until the pass rate falls between 10% and 30%.
  • Rewrite and refine problem parameters iteratively, building an understanding of how the model navigates complex mathematical challenges.
  • Hand off completed tasks to a senior reviewer in your subfield and refine based on their feedback before final submission.


Core Requirements

  • Academic background in Mathematics, Pure or Applied, or an equivalent field.
  • At least 2 years of hands-on experience in mathematics research, applied work, or teaching.
  • Solid Python skills, applied to writing and validating computational solutions.
  • Capacity to build problems that cannot be solved without specialized mathematical software.
  • Excellent written and verbal communication skills in English.
  • Ability to work independently in a remote, fast-paced environment.


Nice-to-Have

  • Working knowledge of one or more domain-specific mathematical tools, including but not limited to Z3, cvc5, Macaulay2, Singular, CasADi, IPOPT, SDPB, G6K, fpylll, or SageMath, or a demonstrated ability to get up to speed independently.
  • Prior exposure to how frontier AI models approach complex scientific problems.
  • Knowledge spanning more than one area of mathematics, such as algebra, optimization, formal verification, or computational number theory.
  • Familiarity with containerized or sandboxed Linux execution environments.


The icing on the Kake:

💵 Competitive Pay in USD – Work globally, get paid globally.

🏡 Fully Remote – Simply put, we trust you.

🌱 Better Me Fund – We invest in your personal growth and passions.

❤️ Compassion is Badass – Join a community that invests in social good.


Please Note: Due to the high volume of applications, only shortlisted candidates will be contacted.

Jobs in Estonia   »   Jobs in Estonia   »   Mathematician with Python - AI Trainer - Remote

More jobs