AI/ML Engineer

No-code drag-and-drop tool to design, test, and deploy AI workflows leveraging Large Language Models for business process automation.
$100,000 - $200,000
Machine Learning
Mid-Level Software Engineer
Hybrid
11 - 50 Employees
3+ years of experience
AI · Enterprise SaaS

Description For AI/ML Engineer

Stack AI is an innovative startup revolutionizing the AI implementation landscape with their no-code drag-and-drop platform for AI workflows. Founded by two MIT PhDs and backed by Y Combinator and Gradient Ventures, the company has shown impressive traction with over 35,000 users and 300+ paying customers in just 8 months. They're seeking a talented AI/ML Engineer to join their four-engineer team.

The role focuses on advancing their AI capabilities, particularly in retrieval-augmented generation (RAG) and embedding techniques. You'll be working at the cutting edge of AI technology, integrating various AI models from providers like OpenAI, Anthropic, and Google, while ensuring seamless connection with common data sources and SaaS platforms.

This position offers a unique opportunity to work directly with founders and make significant impact in a fast-growing startup. The ideal candidate will combine technical expertise in machine learning with strong problem-solving and communication skills. You'll be involved in everything from core AI development to customer support, embodying the company's values of action-oriented decision making, transparency, and bold thinking.

The company's platform enables businesses to build sophisticated AI applications including chatbots, document processing systems, and database query tools. As an AI/ML Engineer, you'll be instrumental in expanding these capabilities and helping democratize access to Large Language Models.

Working at Stack AI means joining a team that's passionate about making AI accessible to everyone while maintaining technical excellence. The role offers competitive compensation ($100K-$200K) and the flexibility of working from major tech hubs including San Francisco, Boston, New York, or Los Angeles.

Last updated 9 hours ago

Responsibilities For AI/ML Engineer

  • Enhance retrieval-augmented generation (RAG) capabilities
  • Work on embedding generation techniques
  • Keep up with latest AI model developments
  • Collaborate with backend and frontend teams
  • Participate in customer support
  • Guide end users in crafting use cases with suitable models

Requirements For AI/ML Engineer

Python
  • Bachelor's or Master's degree in Computer Science, Machine Learning, AI, or related fields
  • Experience in machine learning and AI, focusing on generative models
  • Knowledge of retrieval-augmented generation pipelines
  • Strong proficiency in Python and ML frameworks (TensorFlow/PyTorch)
  • Experience with embedding techniques
  • Excellent problem-solving skills
  • Strong communication and collaboration skills

Interested in this job?

Jobs Related To Stack AI AI/ML Engineer

Software Dev Engineer II, Amazon

Software Engineer II position at Amazon working on foundational LLM development for Amazon Stores, focusing on e-commerce applications

Cloud Support Engineer (Big Data / AI ML), Support Engineering

Cloud Support Engineer role at AWS focusing on Big Data and AI/ML support, combining technical expertise with customer service in a hybrid work environment.

ADC Engineer II, AI/ML ADC

ADC Engineer II position at Amazon focusing on AI/ML solutions in isolated cloud environments for government customers.

ADC Engineer II, AI/ML ADC

ADC Engineer II position at Amazon Web Services, focusing on AI/ML solutions in secure cloud environments, requiring TS/SCI clearance and strong technical expertise.

Software Development Engineer, Sponsored Products Off-Search Sourcing and Relevance

Software Engineer role at Amazon Advertising focusing on ML-driven ad relevance systems with competitive compensation and benefits.