Artificial Intelligence Engineer

Socket.dev; San Francisco (CA); On-site; USD 140,000 - 230,000; Full time

Benefits offered by this job

PTO 20 days per year

Parental Leave

Rewards & Recognition

Insurance Benefits

Job summary

Innovaccer is seeking an AI Engineer to join its growing AI team and build production-grade AI systems that solve complex problems at scale. You will design, develop, and deploy AI-powered applications including LLM-based solutions and RAG-powered workflows.

The role spans end-to-end development from experimentation to production deployment, requiring strong Python, PyTorch, and systems thinking. You will collaborate with product and data teams and communicate with non-AI stakeholders.

Qualifications

• MS or PhD in Computer Science, ML, or related field; exceptional BS with research will be considered.
• First-author publications at NeurIPS/ICML/ICLR/ACL/EMNLP or meaningful open-source ML contributions.
• Fine-tuned open-weight models; explain choice of PE vs full fine-tuning.
• Strong Python and PyTorch skills; familiarity with training/serving stacks.
• Experience with multi-GPU training; understanding sharding strategies.
• Evidence you can finish projects and ship models.

Responsibilities

• Take ideas from concept to prototype to production.
• Build model layers and compose models into working systems with product-level accuracy.
• Write real code in Python and PyTorch; optimize training runs.
• Design experiments and report results that test hypotheses.
• Measure and evaluate rigorously; question seemingly good results.
• Explain AI work to non-engineers and clinicians.

Skills

Python

PyTorch

Fine-tuning

Experimentation

Communication

Multi-GPU training

Education

MS/PhD in CS/ML

Exceptional BS with research

Tools

HuggingFace

FSDP/DeepSpeed

vLLM or SGLang

Job description

About the Role

We are looking for an AI Engineer to join our growing AI team and help build intelligent, production-grade AI systems that solve complex problems at scale.

In this role, you will work closely with product, engineering, and data teams to design, develop, and deploy AI-powered applications, including LLM-based solutions, AI agents, retrieval-augmented generation (RAG), and intelligent automation workflows.

The ideal candidate is hands-on, highly curious, and comfortable working across the full AI development lifecycle—from experimentation and prototyping to production deployment and optimization.

A Day in the Life

• You take an idea from paper to prototype to production. If you have only ever done one of those three, this role will stretch you, and we are fine with that if the rest is strong.
• You can build the model layer of a real product, not just a model. That means choosing model sizes, composing several models into a working system, and holding a product-level accuracy bar.
• You write real code. Python fluently, PyTorch fluently, and enough systems sense to know why your training run is slow.
• You design experiments. You state the hypothesis, run the ablation, and report the result that disagrees with you.
• You measure things. You are suspicious of results that look good, and you build the eval before you build the model.
• You read current research and can tell the difference between a technique that will hold up and one that will not.
• You explain your work to people who are not AI engineers, including clinicians and operators who will tell you when your output is wrong.

What You Need

• MS or PhD in Computer Science, Machine Learning, or a related quantitative field. Exceptional BS candidates with substantial research or open-source work will be considered.
• Depth beyond coursework: first-author publications at NeurIPS, ICML, ICLR, ACL, EMNLP, or similar; meaningful open-source ML contributions; a research internship at an AI lab; or models you trained and shipped that people actually used.
• You have fine-tuned an open-weight model yourself, understand the difference between parameter-efficient and full fine-tuning, and can explain why you chose one.
• Strong Python and PyTorch. Familiarity with the current training and serving stack (HuggingFace, FSDP or DeepSpeed, vLLM or SGLang, or equivalents).
• Some exposure to multi-GPU training, even at lab scale. You should know what a sharding strategy is and why it matters.
• Evidence you can finish things.

We offer competitive benefits to set you up for success in and outside of work.

Here's What We Offer

• Generous PTO Benefits: Enjoy PTO benefit accrual of 20 days per year.
• Parental Leave: Experience one of the industry's best parental leave policies to spend time with your new addition.
• Rewards & Recognition: Unlock your potential and be rewarded generously with both monetary incentives and widespread recognition for your dedication and outstanding performance. Unlock your potential and be rewarded generously with both monetary incentives and widespread recognition for your dedication and outstanding performance.
• Insurance Benefits: We offer medical, dental, and vision benefits along with 100% company-sponsored short and long-term disability and basic life insurance. Legal aid and pet insurance options are available at a discounted rate.

Innovaccer is an equal opportunity employer. We celebrate diversity, and we are committed to fostering an inclusive and diverse workplace where all employees, regardless of race, color, religion, gender, gender identity or expression, sexual orientation, national origin, genetics, disability, age, marital status, or veteran status, feel valued and empowered.

About Innovaccer

Innovaccer activates the flow of healthcare data, empowering providers, payers, and government organizations to deliver intelligent and connected experiences that advance health outcomes. The Healthcare Intelligence Cloud equips every stakeholder in the patient journey to turn fragmented data into proactive, coordinated actions that elevate the quality of care and drive operational performance. Leading healthcare organizations like CommonSpirit Health, Atlantic Health, and Banner Health trust Innovaccer to integrate a system of intelligence into their existing infrastructure- extending the human touch in healthcare. For more information, visit www.innovaccer.com.

Check us out on YouTube, Glassdoor, LinkedIn, Instagram, and the Web.