Full Stack AI Engineer

The Full Stack AI Engineer plays a critical role in making modeling and simulation accessible to non-data scientists. You will be designing, developing, and deploying AI-driven applications to make our software more accessible which will have a direct impact on the number of organizations we are able to serve. This position plays a critical role in our product roadmap and will directly contribute to the company's success and growth. Ideal candidates exhibit a high willingness to experiment and empathy for users.

About Epistemix

Epistemix empowers organizations to make smarter decisions by simulating real-world outcomes using synthetic populations. Our platform enables clients in healthcare, insurance, and government to analyze complex systems and human behavior, delivering actionable insights for critical challenges.

Our mission is to revolutionize decision-making through the use of synthetic populations and simulation, creating a positive impact on industry, government, and academia. Having secured Series A funding, we’re entering an exciting phase of growth and innovation. Join us to combine cutting-edge technology with a purpose-driven mission to make the world a better place.

Responsibilities

• Execute: Craft clean, testable, and maintainable code to enable AI-generated agent-based models.

• Deliver: Own the software from requirements development through deployment and maintenance that enable decision makers to generate agent-based models that address critical business questions and data scientists to build agent-based models more quickly that answer the questions of decision makers.

• System Architecture: Design, build, test, and deploy a scalable system architecture so that AI-generated models can be validated by data scientists and deliver results back to decision makers quickly.

• Strategy: Own the engineering solution and collaborate with internal teams to ensure alignment with company strategy.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field (or equivalent experience).

• 3+ years of experience developing AI/ML applications in production environments.

• Proven track record of working with LLMs, NLP models, or AI-driven systems.

• Experience designing and optimizing high-performance, scalable APIs.

• Strong problem-solving skills and ability to work in a fast-paced environment.

• Must be legally authorized to work in the United States and not require employer sponsorship now or in the future.

Required Skills

• Python – Advanced proficiency in writing clean, efficient, and scalable code.

• Pydantic – Strong experience in data validation, serialization, and structured model definition.

• LLM Evaluation – Ability to assess model performance, optimize outputs, and fine-tune AI behavior.

• Prompt Optimization – Expertise in crafting, refining, and iterating prompts for optimal AI performance.

• SQLAlchemy – Hands-on experience with database modeling, ORM techniques, and performance tuning.

• FastAPI – Proven ability to develop and maintain APIs with FastAPI for AI-driven applications.

Desired Skills

• 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.

• Background in MLOps practices for automating AI model deployment and monitoring.

Why Join Epistemix?

By joining Epistemix, you will become part of a collaborative and quickly growing team that values curiosity and creativity. We are fully remote, with team members in both the United States and Europe. Benefits include incentives such as our stock option program, flexible time off, eligibility for participating in the Epistemix Health and Welfare Program for employees in the United States, and the opportunity to apply your skill set to make an impact.