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.
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
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
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
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
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.