Collaborate on research projects in generative AI and reinforcement learning, design experiments, create datasets, and contribute to publications.
Our mission is to enable hardware deployment at the speed of software development. We are working towards automatic code transpilation and optimization for any hardware application.
In this role, you will collaborate with a small team of talented researchers on ambitious, greenfield projects in generative AI and reinforcement learning.
Core responsibilities:- Design, execute, and analyze experiments with a high degree of independence
- Contribute to core models and frameworks
- Create high-quality datasets (both in-the-wild and synthetic)
- Perform literature reviews and implement new techniques from papers
- Contribute to publications, present at conferences and workshops, etc.
An incomplete list of current and near-term research directions:
- Contrastive representation learning
- Steerability and guided decoding
- Tractable probability models
- Code-specific architectures
- LLM fine-tuning, post-training, RLHF
Requirements
- Ph.D. in Computer Science or a closely related field
- Prior LLM research experience
- Comfortable programming in Python and familiar with frameworks such as PyTorch and HuggingFace
- Publications at peer-reviewed conferences such as NeurIPS, ICLR, ICML, etc.
- Experience with large-scale LLM training, particularly in a distributed computing environment
Benefits
- Competitive salary
- Health care plan (medical, dental, and vision)
- Retirement plan (401k, IRA) with employer matching
- Unlimited PTO
- Flexible hybrid work arrangement
- Relocation assistance for qualifying employees
Top Skills
Huggingface
Python
PyTorch
Similar Jobs
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
The Staff AI/ML Engineer will advance machine learning for autonomous driving by researching and prototyping techniques, collaborating with teams, and publishing findings.
Top Skills:
PythonPyTorch
Information Technology • Software
As a Research Engineer, you will optimize AI performance, focusing on kernel development for large-scale ML workloads, profiling bottlenecks, and collaborating with teams to enhance model training and inference.
Top Skills:
AmdArmAws TrainiumCudaGoogle TpuHipHpcIntelPtxQualcommTriton
.png)
Artificial Intelligence • Big Data • Software • Analytics • Business Intelligence • Big Data Analytics
The AI Engineer will focus on building AI experiences, deploying models, and enhancing data workflows, working on greenfield applications. Key responsibilities include feature development, architecture collaboration, and shipping impactful product experiences.
Top Skills:
AIJavaScriptMachine LearningMlopsPythonTypescript
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
