Skip to content
GranicaGranicaMountain View, CA

Applied AI Research Engineer

Research Engineer building scalable ML systems and pipelines for Large Tabular Models on enterprise structured data. Implement algorithms, optimize training/inference, develop benchmarks, and translate research ideas (led by Stanford Prof. Andrea Montanari) into production.

Salary not listed
On-siteML Engineering

About the role

What You'll Work On

  • Build scalable training, evaluation, and inference pipelines for machine learning systems.
  • Implement and optimize algorithms for structured and tabular data.
  • Develop benchmarks, datasets, and evaluation frameworks for new research ideas.
  • Improve training efficiency, memory usage, and inference performance.
  • Prototype new ML systems and rapidly validate research ideas.
  • Collaborate closely with Prof. Andrea Montanari and Granica's research team to translate research into production systems.

What We're Looking For

  • BS, MS, or PhD in Computer Science, Machine Learning, Mathematics, or a related field.
  • Strong software engineering and machine learning fundamentals.
  • Experience building production ML systems or ML infrastructure.
  • Hands-on experience with PyTorch or JAX.
  • Strong programming skills in Python.
  • Experience developing evaluation frameworks, ML pipelines, or distributed systems.
  • Ability to translate research ideas into reliable, production-quality software.
  • Experience with representation learning, structured or tabular data, probabilistic modeling, distributed training, or ML systems optimization is particularly relevant.

Bonus

  • Experience working closely with research teams.
  • Experience optimizing training or inference at scale.
  • Experience with CUDA, C++, or Rust.
  • Contributions to open-source ML systems.
  • Publications or research experience in machine learning.

Compensation & Benefits

  • Competitive salary, meaningful equity, and performance bonus for top performers.
  • 401(k) with company match, comprehensive health coverage, and unlimited PTO.
  • Daily catered meals in our Mountain View office.
  • Support for research, publication, and conference participation.

Skills

PyTorchJAXPythonCUDAC++RustDistributed TrainingMl PipelinesRepresentation LearningProbabilistic Modeling

Similar roles

ML Engineering jobs
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k
Hybrid5+ YOEML Engineering
Liftoff

Machine Learning Engineer

LiftoffCalifornia

Machine Learning Engineer building statistical models, optimization systems, and experiments for mobile ad tech economics on the Revenue Engine team. Requires PhD in CS/ML/Economics and industry experience applying ML or economics at scale.

215k – 275k
RemoteML Engineering