Skip to content
GranicaGranicaSan Francisco, CA

Research Scientist – Tabular & Structured Machine Learning

Conduct research and build foundational ML models for structured and tabular data, combining statistical learning theory, probabilistic modeling, and large-scale systems. Requires a PhD and strong experience in tabular/relational ML.

160k – 250k
On-siteEntry levelAI Research

About the role

What You’ll Build and Research

  • Invent and prototype algorithms that advance the foundations of machine learning for structured and tabular data
  • Develop new representation learning techniques and information models for large enterprise datasets
  • Build adaptive learners combining statistical learning theory, probabilistic modeling, and large-scale systems optimization
  • Contribute to the development of large tabular models and structured foundation models
  • Design architectures integrating relational, symbolic, and neural learning components
  • Research and implement methods for dataset compression, selection, and representation to improve learning efficiency
  • Develop cost models and optimization frameworks for large-scale structured learning systems
  • Collaborate closely with the Granica research group led by Prof. Andrea Montanari (Stanford) and with systems engineers
  • Rapidly prototype new algorithms and evaluate them on real enterprise datasets
  • Publish and contribute to the broader research community shaping the future of structured AI and efficient ML systems

What You’ll Bring

  • PhD in Machine Learning, Statistics, Computer Science, Applied Mathematics, or a related field
  • Research experience related to structured, relational, or tabular data
  • Experience in one or more of the following areas:
    • Tabular or relational machine learning
    • Representation learning for structured data
    • Statistical learning theory or generalization
    • Probabilistic modeling or Bayesian inference
    • Optimization for machine learning
    • Scalable or distributed ML systems
  • Experience working with structured datasets or relational data systems
  • Strong grounding in statistics, optimization, information theory, or probabilistic inference
  • Hands-on experience with PyTorch, JAX, or TensorFlow
  • Strong programming skills in Python or Rust
  • Demonstrated ability to translate theoretical ideas into working systems or prototypes
  • Curiosity about how structure and relational information enable new forms of learning and reasoning
  • A pragmatic research mindset: you value elegant ideas but also ship systems that work at scale

Bonus

  • Research in tabular machine learning, relational representation learning, or structured data modeling
  • Experience building large-scale ML infrastructure or distributed training systems
  • Familiarity with data systems, query engines, or dataset optimization pipelines
  • Publications at top venues such as NeurIPS, ICML, ICLR, COLT, KDD, AAAI
  • Contributions to open-source ML systems or research-to-production tooling

Compensation & Benefits

  • Competitive salary, meaningful equity, and substantial bonus for top performers
  • Flexible time off plus comprehensive health coverage for you and your family
  • Support for research, publication, and deep technical exploration

Skills

Machine LearningPyTorchJAXTensorFlowPythonRustTabular Machine LearningRepresentation LearningProbabilistic ModelingStatistical Learning Theory

Similar roles

AI Research jobs
Snowflake

Post-Doctoral Researcher

SnowflakeBellevue, WA

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 – 220k
HybridEntry levelAI Research
Nuro

Machine Learning Research Scientist: Generative Modeling for Planning

NuroMountain View, CA

Develops state-of-the-art generative models like diffusion and flow-matching for autonomous planning in self-driving tech. Requires PhD or MSc with 2-3 years experience in generative modeling for robotics, strong Python/C++ skills, and top research publications.

160k – 241k
On-site2+ YOEAI Research
Snowflake

AI Research Scientist, New Grad – Agents & Reinforcement Learning

SnowflakeBellevue, WA

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 – 230k
On-siteEntry levelAI Research
Greptile

Research Engineer

GreptileSan Francisco, CA

Research Engineer building autonomous AI agents that review GitHub PRs, catch bugs, learn coding standards, and improve signal-to-noise for 7,000+ customers. Requires CS degree, research experience with ML/LMs/agents, strong programming, and product intuition.

180k – 280k
On-siteEntry levelAI Research
Nuro

ML Research Scientist, Prediction & Smart Agents

NuroMountain View, CA

Build state-of-the-art ML models to predict traffic behavior for autonomous driving, using generative sequence modeling and controllable agents for planning and simulation. Requires PhD preferred, 2+ years deploying ML systems, and expertise in PyTorch and robotics ML.

194k – 291k
On-site2+ YOEAI Research