Staff AI Engineer responsible for generative AI solutions for conversational products and operational efficiency at Stone Co. Engaging in technical leadership and collaboration between science and engineering.
About the role
- AI Engineer developing agentic AI workflows to enhance drug discovery processes using advanced machine learning. Collaborating with medicinal chemists to deliver effective solutions.
Responsibilities
- Own the implementation of end-to-end agentic AI workflows, from data ingestion and schema design to prompting, context management, tool creation, and multi-agent orchestration
- Partner closely with medicinal chemists to prototype, test, and refine AI-assisted workflows that improve scientific quality, speed, and confidence
- Improve real user use cases by incorporating feedback from internal users (and external customers post-launch), translating insights into product and system improvements
- Support automated integration of customer assay and experimental data into Indy environments, ensuring reliability and scalability
- Identify and implement product improvements or new features based on hands-on collaboration with scientific, product, and engineering stakeholders
- Contribute to backend and system architecture decisions as the platform evolves from single-user workflows to scalable, multi-user production systems
Requirements
- 4+ years of experience building backend and/or data systems using Python in production environments
- Experience shipping at least one user-facing system that integrates LLMs (agentic workflows are a strong plus)
- Strong understanding of backend systems, data flows, and production infrastructure
- Comfort working with data engineering concepts such as schema design, pipelines, and orchestration (Dagster, Airflow, or similar tools)
- Ability to collaborate deeply and effectively across disciplines, including working directly with scientists and translating domain-specific needs into robust technical solutions
- Comfortable owning ambiguous problems, building from first principles, and iterating quickly in a startup environment
Benefits
- competitive salary and equity-based compensation
- comprehensive healthcare benefits (including dental and vision)
- opportunity to grow alongside a rapidly scaling company
Job title
Job type
Experience level
Salary
Degree requirement
Tech skills
Location requirements
Forward Deployed AI Engineer assisting enterprises with Adaptive Engine deployments for generative AI. Collaborating with technical teams to enhance customer success and model performance.
AI Engineer responsible for building and deploying fullstack AI features for clinical applications. Collaborate closely with product leads at Heidi to enhance healthcare delivery through technology.
AI Engineer developing innovative AI solutions for stakeholder intelligence at TSC. Designing data pipelines and optimizing ML models in a collaborative environment.
AI Engineer developing intelligent data pipelines and multi - agent workflows in Vietnam. Collaborating with the Singapore hub to deliver impactful stakeholder intelligence results.
AI Platform/ Model Developer at Mars leveraging AI to enhance North America Supply Chain efficiency and resilience. Collaborating with various teams to design and implement scalable AI capabilities.
Principal AI Engineer leading platform engineering and AI enablement initiatives at Humana, driving strategy for AI tools and products while collaborating with cross - functional partners.
AI Engineer at Worldia creating and deploying AI workflows for travel agencies. Collaborating cross - functionally with product, data, and engineering teams to solve concrete problems.
Design and build an agentic AI platform to manage renewable energy certificates and automate related workflows. Collaborate across teams to integrate the platform, optimize user onboarding, and iterate based on feedback and metrics.
AI Engineer - Consultant developing and integrating AI solutions for clients. Working in a hybrid model to support their journey towards data - driven processes.