Skip to content

Machine Learning Researcher - Springtail

150k – 300kEmeryville, CAAI ResearchHybrid
Summary

Conducts fundamental research on data-efficient ML architectures, including bootstrapped program synthesis and self-synthesizing learning systems. Requires Master's in ML/math, PyTorch fluency, and research experience.

About the role

Responsibilities

  • Hypothesize, test, and refine means of improving generalization performance of common architectural elements, including different forms of attention. This includes devising controlled datasets to elucidate e.g. learning order & learned representations.
  • Think both mathematically and empirically about problems of runtime inference in gradient-trained networks, with an eye to the extensive literature on statistical learning and an open mind to the many forms of constrained optimization.
  • Contribute to a well-documented and well-instrumented code base that is performant where necessary yet expeditious where experimental throughput demands.

Qualifications and Experience

  • Masters or equivalent in machine learning, mathematics, or equivalent fields (strong candidates from neuroscience are encouraged to apply).
  • Fluency with Pytorch, and familiarity with JAX, CUDA, and/or Triton + their open-source ecosystems.
  • Demonstrated ability do fundamental research.
  • Demonstrated ability to work in teams.
Skills
PyTorchJAXCUDATritonattention mechanismsprogram synthesisgradient-trained networksstatistical learningconstrained optimization
Similar roles at this salary range
All AI Research jobs →
Snowflake

AI Research Scientist, New Grad – Agents & Reinforcement Learning

Conduct research on autonomous AI agents and reinforcement learning to build self-improving systems that reason, code, and learn at scale within the Snowflake Data Cloud. Requires a PhD (or equivalent) and strong expertise in RL and agentic AI.

176k – 230kBellevue, WAAI ResearchOn-siteEntry levelJAXDPO
Together AI

Frontier Agents Intern

Research intern on the Agents team building and aligning frontier AI systems for complex agentic and scientific tasks. Focus on post-training methods, evaluation frameworks, self-learning, and scalable agent infrastructure.

121k – 131kSan Francisco, CAAI ResearchOn-siteEntry levelJAXNLP
Snowflake

Post-Doctoral Researcher

Post-doctoral researcher conducting independent and collaborative AI/ML research focused on high-impact domains like medicine, finance, and law. Requires a recent or imminent PhD and publications in top venues.

160k – 220kBellevue, WAAI ResearchHybridEntry levelJAXRAG
SpotOn

Senior Software Engineer - Python/Typescript

Senior engineer building AI-driven automation systems to replace manual business workflows across operations, sales, and support. Requires 7+ years experience, production Python/TypeScript skills, and 1-2 years building agentic AI systems.

160k – 190kChicago, IL +3AI ResearchHybrid7+ YOEAWSLLMs
Datology AI

Research Engineer

As a Research Engineer, you will conduct and enable cutting-edge research, translating it into the core product pipeline. You will develop and improve state-of-the-art data curation strategies, accelerating research and ensuring product innovation.

180k – 300kRedwood City, CAAI ResearchOn-site4+ YOEML ModelsAI Models