Senior Staff AI Engineer

About the role

We are seeking a highly motivated and talented individual with a passion for AI/ML engineering and agent technologies. In this role, you will unleash your creativity, intelligence, and curiosity to build scalable AI systems that empower researchers and chemists at leading chemical and materials organizations to push the boundaries of innovation. 

As an ML Engineer specializing in LLMs and agent technologies, you will play a critical role in our mission to streamline workflows and provide robust, scalable solutions that support AI/ML capabilities for thousands of researchers worldwide. This role is central to the development of autonomous systems and tools tailored to chemical and materials science applications. 

What you'll do

We are seeking an exceptional ML Engineer with a focus on LLMs and RAG systems. This role prioritizes designing and developing scalable, fault-tolerant AI systems while maintaining a strong focus on domain-specific AI solutions. You will play a critical role in building robust infrastructure to support high-performance applications and tools, enabling seamless data integration and transformation to power AI/ML capabilities in chemical and materials science. 

Scalable AI System Development:

• Design, build, andmaintainscalable, fault-tolerant AI systems leveraging OpenAI and Anthropic models. 

• Develop RAG architectures to ensure efficient, high-performance information retrieval tailored to chemical and materials science. 

• Optimizesystem performance to handle large-scale data and application demands. 

AI Agent Development:

• Build andmaintainintelligent AI agents using modern frameworks. 

• Collaborate with domain experts to refine agent capabilities for specific scientific workflows. 

Data Engineering and Integration:

• Architect andmaintainvector database solutions for efficient data storage and retrieval. 

• Develop pipelines for ingestion, transformation, and storage to enable AI/ML workflows. 

• Collaborate with platform and ML engineers to integrate AI/ML models with backend systems. 

System Reliability and Fault Tolerance:

• Implement robust error-handling, monitoring, and alerting mechanisms to ensure system resilience. 

• Troubleshoot and resolve system bottlenecks and failures. 

CI/CD and Deployment Pipelines:

• Design, implement, andmaintainCI/CD pipelines for AI systems and data workflows. 

• Promote automation and best practices to enhance thedevelopmentlifecycle. 

Adoption of Emerging Technologies:

• Stay informed on the latest trends and tools in AI/ML engineering and agent technologies. 

• Introduce and implementnew technologiesto improve system scalability, data integration, and developer productivity. 

Collaboration and Cross-Functional Teamwork:

• Work closely with AI/ML, data engineering, and platform teams to understand anddeliver ontechnical requirements. 

• Contribute to architectural decisions thatimpactthe overall platform ecosystem. 

You will have

• A strong passion for AI/ML engineering and scalable data systems. 

• Anability to prioritize system scalability and fault tolerance while focusing on innovative AI/ML solutions. 

• A collaborative mindset and excellent communication skills. 

• A commitment to quality and innovation in AI and data engineering. 

Key competencies

• A degree in Computer Science, AI, or a related field with 7+ years of industry experience (Bachelor’s) or 5+ years (Master’s or PhD) in software engineering, emphasizing expertise in building scalable, fault-tolerant AI systems. 

• Advanced knowledge of modern AI frameworks (e.g., LangChain, LangGraph, AutoGen, Crew.ai). 

• Experience with vector databases (e.g., Pinecone, Milvus, Pgvector, ChromaDB). 

• Strong understanding of distributed systems and microservices architecture. 

• Proficiency in REST API development using FastAPI REST Framework or similar tools. 

• Familiarity with cloud platforms (e.g., AWS, GCP, Azure) and containerization technologies (e.g., Docker, Kubernetes). 

• Proven track record of deploying production-grade AI systems. 

• Experience leading technical teams and fostering collaborative environments. 

Preferred/Bonus Points

• Advanced degree in Computer Science, AI, or related fields. 

• Background in chemical and materials science applications. 

• Contributions to open-source AI projects. 

• Experience with fine-tuning and optimizing large language models (LLMs). 

• Expertise in system monitoring and observability tools (e.g., Prometheus, Grafana). 

• Experience with data engineering tools (e.g., Airflow, Delta Lake, Dask). 

• Familiarity with Unix scripting and version control systems (e.g., Git). 

• Prior experience working in AI/ML-focused environments using tools such as Ray, Torch, Kubeflow.