AI/ML Engineer

About us

Stack AI is a no-code drag-and-drop tool to quickly design, test, and deploy AI workflows that leverage Large Language Models (LLMs), such as ChatGPT, to automate any business process.

Our core value is to make it extremely easy to build arbitrarily complex AI pipelines using a visual interface that allows you to connect different data sources with different AI models.

Our customers use Stack AI to build applications such as:

• Chatbots and Assistants: AI agents that interact with users, answer questions, and complete tasks, using your internal data and APIs.
• Document Processing: apps to answer questions, summarize, and extract insights from any document, no matter how long.
• Answer Questions on Databases: connect GPT-like models to databases (such as Notion, Airtable, or Postgres) and ask questions about them.
• Content Creation: generate tags, summaries, and transfer styles or formats between documents and data sources.

Weve already earned the trust of 100,000+ users and 300+ paying customers (yes, even unicorns and public companies!), and were backed by heavyweights like Y Combinator and Google.

Our team is small (6 brilliant engineers), but mighty, and were all about radical transparency, bold thinking, and action-first. If you're ready to ship code that actually gets used and loved, read on.

Your main responsibilities

In this role, you will:

• Enhance our RAG capabilities, making our models smarter and more context-aware.
• Optimize embedding generation techniques for performance and precision.
• Stay on the pulse of AI innovation, and serve as the teams go-to expert on evolving model capabilities.
• Collaborate closely with frontend and backend teams to bring AI features to life.
• Get hands-on with users, understand their needs, help them craft AI-powered solutions, and occasionally jump into customer support like the rest of us.

Requirements for this position

You will be a great fit if you have:

• A Bachelor's or Master's degree in Computer Science, Machine Learning, AI, or a related field.
• Strong Python skills, and hands-on experience with TensorFlow or PyTorch.
• Experience with Machine Learning, especially Generative Models.
• Knowledge of retrieval-augmented generation (RAG) pipelines.
• Understanding of embedding techniques and how they fuel intelligent systems.
• Excellent problem-solving abilities and comfort working in fast-paced environments.
• Great communication and collaboration skills: Were a tight-knit team.

Your compensation

This is your chance to join a rocket ship at liftoff. At Stack AI, you wont just write code, youll shape a category-defining product, used by real customers, in real time. Youll work alongside top-tier engineers, build with a next-level tech, and have a voice in core technical and product decisions.

If youre a 10X builder, a curious mind, and ready to make serious impact with your work: Were waiting for you.

Languages

• Required (fluent): English
• Valuable: Spanish

Valuable skills

Bonus points if you have:

• A PhD or published work in generative AI, embeddings, or RAG.
• Experience deploying large-scale ML systems in production.
• Contributions to open-source AI/ML projects.
• Deep knowledge of GPU architectures and hardware optimization for ML workloads.