Apply on Employer Site

StackAI · 4 months ago

Senior AI/ML Engineer

SF Office - 171 2nd, 4th floor

Full-time

Onsite

Senior Level

$150K/yr - $250K/yr

Stack AI is a no-code platform designed to simplify the creation, testing, and deployment of AI workflows utilizing Large Language Models. In this role, the Senior AI/ML Engineer will enhance retrieval-augmented generation capabilities, optimize embedding generation techniques, and collaborate with teams to integrate AI models into products while maintaining user relationships.

AppsArtificial Intelligence (AI)Developer PlatformSoftware

H1B Sponsor Likely

Responsibilities

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

Qualification

Machine LearningGenerative ModelsPythonTensorFlowPyTorchRetrieval-Augmented GenerationEmbedding TechniquesCuriosityProblem-SolvingCommunication SkillsCollaboration Skills

Required

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 TensorFlow 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

Preferred

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

Company

StackAI

Build and deploy Enterprise-Grade AI Agents.

Founded in 2023

Cambridge, Massachusetts, USA

51-200 employees

https://stack-ai.com

H1B Sponsorship

StackAI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (4)