AI Engineer

About the Role

Day-to-day, the AI Engineer will be involved in crafting clean, testable, and maintainable code to enable AI-generated agent-based models. This requires a deep understanding of AI/ML applications, experience with large language models (LLMs), and the ability to optimize high-performance, scalable APIs. The engineer will work closely with internal teams to ensure that the engineering solution aligns with the company's overall strategy, contributing to a collaborative and dynamic work environment.

Given Epistemix's approach to solving real-world problems through simulation and data-driven modeling, the AI Engineer will be at the heart of innovation, working on projects that have the potential to impact public health, life sciences, insurance, and enterprise strategy. The role demands a high willingness to experiment, empathy for users, and strong problem-solving skills, making it a challenging yet rewarding opportunity for the right candidate.

What You Will Do

• Design, develop, and deploy AI-driven applications to make modeling and simulation more accessible.
• Craft clean, testable, and maintainable code for AI-generated agent-based models.
• Own the software development process from requirements to deployment and maintenance.
• Develop and optimize high-performance, scalable APIs for AI applications.
• Collaborate with internal teams to ensure alignment with company strategy.
• Design and build a scalable system architecture for AI-generated models.
• Work with data scientists to validate AI-generated models and deliver results to decision-makers.
• Continuously evaluate and improve the performance of AI models.
• Develop and implement prompt optimization techniques for LLMs.
• Ensure the security, integrity, and compliance of AI systems.

What We Are Looking For

• Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field.
• 3+ years of experience in developing AI/ML applications in production environments.
• Proven track record of working with LLMs, NLP models, or AI-driven systems.
• Experience with designing and optimizing high-performance, scalable APIs.
• Strong problem-solving skills and the ability to work in a fast-paced environment.
• Advanced proficiency in writing clean, efficient, and scalable Python code.
• Strong experience with Pydantic for data validation, serialization, and structured model definition.
• Ability to assess model performance, optimize outputs, and fine-tune AI behavior.
• Expertise in crafting, refining, and iterating prompts for optimal AI performance.
• Hands-on experience with SQLAlchemy for database modeling, ORM techniques, and performance tuning.
• Proven ability to develop and maintain APIs with FastAPI for AI-driven applications.

Nice to Have

• Experience with vector databases (e.g., Pinecone, Weaviate, FAISS) for efficient AI retrieval.
• Familiarity with Docker & Kubernetes for containerized AI application deployment.
• Knowledge of cloud platforms (AWS, GCP, or Azure) for scaling AI infrastructure.
• Understanding of retrieval-augmented generation (RAG) techniques.

Benefits and Perks

• Competitive salary and benefits package.
• Opportunity to work on projects that impact real-world challenges.
• Collaborative and dynamic work environment.
• Professional development and growth opportunities.
• Flexible remote work arrangements.
• Access to the latest technologies and tools in AI and ML.
• Health insurance and wellness programs.
• Generous PTO policy.
• Remote stipend for home office setup and internet reimbursement.

• Highlight your problem-solving skills: Showcase projects or experiences where you've had to solve complex problems, especially in AI/ML contexts.
• Prepare to talk about your experience with AI/ML applications: Be ready to discuss your experience with LLMs, NLP models, or other AI-driven systems, and how you've applied them in real-world scenarios.
• Emphasize your ability to work in fast-paced environments: Demonstrate how you can handle the pace and pressure of working in a dynamic startup environment.
• Showcase your proficiency in Python and relevant tools: Make sure your portfolio or resume highlights your skills in Python, Pydantic, SQLAlchemy, and FastAPI.
• Be prepared to discuss prompt optimization techniques: Understand how to craft, refine, and iterate prompts for optimal AI performance and be ready to discuss this in an interview.
• Research Epistemix and its mission: Understanding the company's goals and how your role contributes to them will show your interest and motivation for the position.
• Prepare questions about the company culture and team: Asking insightful questions about the work environment and team dynamics can leave a positive impression on the interviewer.