Skip to content
AsteraAsteraEmeryville, CA

Machine Learning Researcher - Springtail

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.

150k – 300k
HybridAI Research

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

AI Research jobs
Distyl AI

Applied AI Researcher, Multi-Agent Systems

Distyl AISan Francisco, CA +1

Develop multi-agent AI architectures for enterprise coordination and collaborative reasoning. Requires research experience in MARL/GNNs, strong prototyping skills, and daily AI tool usage.

150k – 250k
HybridAI Research
Basis

Intelligence Architect

BasisNew York, NY

Designs and authors context, procedures, skills, and system prompts for AI agents performing autonomous accounting tasks. Ensures consistency, monitors performance, and shapes future capabilities through precise language and systems thinking.

150k – 225k
On-siteAI Research
AfterQuery

Research Scientist - Frontier Data

AfterQuerySan Francisco, CA

Designs datasets and evaluation frameworks for frontier AI models, collaborating with top labs to expose failure modes, refine RLHF/RLVR pipelines, and measure data impact on capabilities. Requires strong quantitative skills, familiarity with LLM training, and research experience up to master's level.

150k – 250k
On-siteAI Research
AfterQuery

Research Scientist - Post Training

AfterQuerySan Francisco, CA

Design and run SFT/RL experiments to measure dataset impact on LLM performance, capabilities, and alignment. Collaborate with labs to provide evidence of improvements; requires strong LLM training knowledge and fast experimentation, ideally undergrad/master's research background.

150k – 250k
On-siteAI Research
Astera

Research Scientist: Hierarchical Sensorimotor Perception

AsteraEmeryville, CA

Develops bi-directional/recurrent architectures for hierarchical sensorimotor perception inspired by brain principles to advance AGI. Requires PhD-level expertise in deep learning frameworks like PyTorch/JAX, theoretical rigor, and passion for biological intelligence.

150k – 300k
HybridAI Research