Senior AI​/ML Engineer

Join to apply for the Founding Engineer - AI/ML role at StackAI

Join to apply for the Founding Engineer - AI/ML role at StackAI

Get AI-powered advice on this job and more exclusive features.

This range is provided by Stack

AI. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

$/yr - $/yr

About Stack AI

• Vision:
  Democratize access to Large Language Models, so that anyone can build AI-powered applications that generate positive impact to the world.
• Product: A user-friendly and intuitive No-Code platform integrating best AI models (OpenAI, Anthropic, Google, open source), most common data sources (One Drive, Google Drive, Airtable, Notion) and SaaS (Zapier, Hub Spot, etc.). We do the heavy-lifting for you, so you can focus on the product and business.
• Traction: over 100,000 users and 300+ paying customers (including public companies and unicorns!).
• Team:
  Founded by 2 MIT PhDs, backed by major investors such as Y Combinator and Google, we are now a 6-engineer team.
• Values:
  Default to action, radical transparency, boldness, and strong opinions loosely-held.

• Enhance our retrieval-augmented generation (RAG) capabilities, ensuring our models effectively integrate and utilize diverse data sources.
• Work on embedding generation techniques, optimizing them for accuracy and efficiency.
• Keep abreast of the latest developments in AI models, their evolving capabilities, and the complexities of integration. Position yourself as the go-to expert within the team, guiding our end users in crafting their use cases with the most suitable models.
• Collaborate with backend and frontend teams to integrate AI models into products and services.
• We're all about our users - listen to their needs, help them, build strong relationships! Expect to dive into customer support now and then as everyone else in the team.

• Bachelor's or Master's degree in Computer Science, Machine Learning, AI, or related fields.
• Proven experience in machine learning and AI, with a focus on generative models.
• Knowledge of retrieval-augmented generation pipelines and their implementation.
• Strong proficiency in Python and popular machine learning frameworks like Tensor Flow or PyTorch.
• Experience with embedding techniques and understanding of their applications in AI models.
• Excellent problem-solving skills and the ability to work in fast-paced environments.
• Strong communication and collaboration skills.

• PhD in a relevant field.
• Published research or projects in the field of generative AI, embeddings, or retrieval-augmented generation.
• Experience with large-scale ML deployments in a production environment.
• Contribution to open-source AI/ML projects.
• In-depth knowledge of generative AI hardware, including an understanding of GPU architectures, requirements, and optimization techniques for AI model deployment.

• Seniority level
  
  Mid-Senior level

• Employment type
  
  Full-time

• Job function
  
  Engineering and Information Technology
• Industries Software Development

Referrals increase your chances of interviewing at Stack

AI by 2x

Apply BELOW