AI Engineer

Epistemix

Pittsburgh (Allegheny County)

On-site

USD 90,000 - 120,000

Full time

14 days+

Application generator

Stand out for this role — generate a tailored resume and cover letter in about a minute.

Get past ATS filters

Benefits offered by this job

Equity & Incentives

Flexible Time Off

Health, Welfare and 401(k) Programs

Meaningful Impact

Job summary

A dynamic data science firm is seeking a Full Stack AI Engineer to design and deploy AI-driven applications. This remote role, based in the U.S., requires a strong background in Python, Pydantic, and AI models. As a critical member of the team, the engineer will develop scalable architectures to enhance modeling and simulation accessibility. Candidates should have experience in production environments and a commitment to impact-driven projects. Flexible benefits and equity participation are offered.

Qualifications

3+ years of experience developing AI/ML applications in production environments.
Proven track record of working with LLMs, NLP models, or AI-driven systems.
Strong problem-solving skills and ability to work in a fast-paced environment.

Responsibilities

Craft clean, testable, and maintainable code to enable AI-generated models.
Design, build, test, and deploy a scalable system architecture.
Collaborate with internal teams to align with the company strategy.

Skills

Python

Pydantic

LLM Evaluation

Prompt Optimization

SQLAlchemy

FastAPI

Education

Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field

Tools

Docker

Kubernetes

Cloud platforms (AWS, GCP, Azure)

Job description

Join to apply for the Full Stack AI Engineer role at Epistemix.

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 helps organizations forecast outcomes and manage risk by modeling how people behave. Our platform consists of two core products: Populus, which provides access to realistic, high-resolution synthetic population data, and Polaris, which enables scenario planning through advanced data science and simulation. Together, they empower customers to evaluate the potential impact of strategies before deploying them in the real world.

Organizations across healthcare, consumer industries, insurance, and government use Epistemix to reduce uncertainty, optimize decisions, and accelerate time to value. Whether estimating total addressable market, testing public health interventions, or forecasting behavior change, our tools help teams make confident decisions when pre-existing data may not exist.

Leading up to our Series B financing, we are hiring key roles onto the team and 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

Craft clean, testable, and maintainable code to enable AI-generated agent‑based models.
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.
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.
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.

Nice To Have Experience

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?

Equity & Incentives – Participation in our stock option program.
Flexible Time Off – Autonomy to manage your schedule and work‑life balance.
Health, Welfare and 401(k) Programs – Eligibility for benefits (for U.S. employees).
Meaningful Impact – Apply your creative talents to revolutionize data‑driven decision‑making and make a real‑world difference.

This is a remote position open to applicants located in the United States. Candidates must possess the legal right to work in their intended work location, as we are currently unable to sponsor or transfer employment visas for any country, including the United States.

Seniority Level