AI Engineer

BayRockLabs; Newark (CA); Hybrid; USD 110,208 - 123,984; Full time

Job summary

A leading product engineering firm in Newark, California is seeking a skilled AI Engineer to design, build, and integrate advanced AI systems powered by large language models (LLMs). The ideal candidate will have strong expertise in Python, experience with LangGraph/LangChain, and proficiency in FastAPI for developing production-grade APIs. This hybrid role offers competitive pay and the opportunity to work on innovative projects in a collaborative environment.

Qualifications

• Strong hands-on experience with LLM-powered systems (e.g., GPT, Claude, LLaMA, Mistral).
• Deep expertise in Python for AI/ML development.
• Proven experience with LangGraph/LangChain for agent workflow orchestration.

Responsibilities

• Design and build advanced AI systems powered by large language models (LLMs).
• Develop and integrate agent workflows using LangGraph/LangChain frameworks.
• Implement production-grade APIs with FastAPI to support scalable deployments.

Skills

LLM-powered systems

Python

LangGraph/LangChain

FastAPI

ML lifecycle management

Cloud platforms (Azure, AWS, or GCP)

Problem-solving skills

Tools

MLOps tools (MLflow, Kubeflow)

Vector databases (Pinecone, Weaviate, FAISS)

Job description

About BayRock Labs

At BayRock Labs, we pioneer innovative tech solutions that drive business transformation. As a leading product engineering firm based in Silicon Valley, we provide full-cycle product development, leveraging cutting-edge technologies in AI, ML, and data analytics. Our collaborative, inclusive culture fosters professional growth and work-life balance. Join us to work on ground-breaking projects and be part of a team that values excellence, integrity, and innovation. Together, let's redefine what's possible in technology.

We are seeking a highly skilled AI Engineer with strong hands‑on experience in building and integrating LLM-powered systems. The ideal candidate will have deep expertise in Python, practical experience with LangGraph/LangChain for agent workflows, and proficiency in FastAPI for developing production-grade APIs. This role involves designing, deploying, and scaling intelligent systems that leverage large language models to deliver enterprise‑ready solutions.

Key Responsibilities

• Design and build advanced AI systems powered by large language models (LLMs).
• Develop and integrate agent workflows using LangGraph/LangChain frameworks.
• Implement production‑grade APIs with FastAPI to support scalable deployments.
• Collaborate with cross‑functional teams (data scientists, ML engineers, product managers) to deliver end‑to‑end AI solutions.
• Optimize performance, reliability, and scalability of AI‑driven applications.
• Stay current with emerging AI/ML technologies and contribute to innovation initiatives.

Required Skills & Experience

• Strong hands‑on experience with LLM-powered systems (e.g., GPT, Claude, LLaMA, Mistral).
• Deep expertise in Python for AI/ML development.
• Proven experience with LangGraph/LangChain for agent workflow orchestration.
• Proficiency in FastAPI for building robust, production‑ready APIs.
• Solid understanding of ML lifecycle management, including deployment, monitoring, and scaling.
• Experience with cloud platforms (Azure, AWS, or GCP) for AI/ML workloads.
• Strong problem‑solving skills and ability to work in fast‑paced environments.

Preferred Qualifications

• Experience with vector databases (e.g., Pinecone, Weaviate, FAISS) for retrieval‑augmented generation.
• Familiarity with MLOps tools (MLflow, Kubeflow, or similar).
• Knowledge of prompt engineering and fine‑tuning LLMs.
• Prior work in enterprise AI applications (chatbots, copilots, automation systems).

The pay range for this role is:

80 - 90 USD per hour (Hybrid (Newark, California, US))