Research Scientist
Research Scientist developing novel diffusion models and generative algorithms for Large Tabular Models (LTMs) on enterprise data. Requires PhD and strong track record in generative ML; experience with diffusion models and PyTorch/JAX preferred.
About the job
What You'll Work On
- Develop novel diffusion models and generative learning algorithms.
- Research new representation learning techniques for Large Tabular Models.
- Design efficient training methods for large-scale generative models.
- Prototype and evaluate new generative modeling approaches.
- Design rigorous experiments and benchmarks to measure model quality and efficiency.
- Collaborate closely with Prof. Andrea Montanari and Granica's research team to translate research into production systems.
What We're Looking For
- PhD in Machine Learning, Computer Science, Statistics, Applied Mathematics, or a related field.
- Strong research record in generative machine learning.
- Experience developing new generative models or learning algorithms.
- Hands-on experience with PyTorch or JAX.
- Strong programming skills in Python.
- Ability to turn research ideas into working systems.
- Experience with diffusion models, score-based generative modeling, representation learning, probabilistic modeling, or scalable ML systems is particularly relevant.
Bonus
- Research applying diffusion models beyond traditional vision tasks.
- Publications at NeurIPS, ICML, ICLR, CVPR, ICCV, ECCV, or related venues.
- Open-source or production ML systems experience.
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
Diffusion Models, Generative Models, PyTorch, JAX, Python, Representation Learning, Probabilistic Modeling, Scalable Ml
Similar jobs
AI Research jobs
Improbable
Remote
Conduct applied research on AI agents, designing experiments and evaluation systems to improve reliability, context retention, and multi-step task completion. The role requires strong AI/ML research, engineering, experimental design, and communication skills.
Anthropic
San Francisco, CA
Research Engineer building large-scale AI capability evaluations, telemetry, data pipelines, and analysis tools for Anthropic’s Takeoff Intel team. The role requires hands-on large language model experimentation, rapid prototyping, data expertise, and strong research collaboration.
Sardine
United States
Conduct applied research on foundation models for fraud detection using large-scale behavioral and financial-risk data. The role spans experimentation, evaluation, production deployment, and cross-functional work on model governance, requiring 4+ years of applied ML experience and strong Python and SQL skills.
OpenAI
San Francisco, CA
Researcher or engineer focused on designing, evaluating, and productionizing oversight systems and safety mitigations for autonomous AI agents. The role requires strong systems or security reasoning, threat-modeling ability, and experience building practical evaluations and controls.
OpenAI
San Francisco, CA
Researcher focused on training and evaluating frontier AI agents, mining incidents, and building scalable safety measurement systems. The role requires strong research or ML engineering execution, quantitative judgment, and the ability to own ambiguous projects end to end.