AI/ML Engineer

Flexcompute is transforming how the world designs electromagnetic and photonic systems. Tidy3D, our flagship EM simulation platform, is the industry's fastest, most scalable GPU-native solver, empowering companies in semiconductors, photonics, AR/VR, quantum, RF systems, sensors, and advanced computing to simulate complex EM behavior orders of magnitude faster than legacy CPU tools.

Our company was founded by world-renowned leaders in simulation technology from Stanford University and MIT. Backed by top VC firms, we are poised to disrupt the billion-dollar engineering simulation industry with our fast-growing trajectory.

Location: Remote (EU timezone preferred)

We are looking for an AI/ML engineer to build and scale our AI-powered simulation assistant, which combines LLM orchestration with semantic search in a domain where precision and technical accuracy matter. You will own the full stack from embeddings pipelines to production inference, working closely with physicists and engineers to ground AI outputs in scientific correctness.

Responsibilities

• Design and maintain LLM-based agentic systems for physics simulation workflows
• Build semantic search and retrieval pipelines over technical documentation and simulation data
• Develop embedding pipelines: chunking strategies, vector stores, retrieval evaluation
• Deploy and operate containerized ML services on AWS (ECS, Lambda, S3)
• Optimize LLM inference costs, latency, and quality at scale
• Integrate AI capabilities into IDE extensions (VS Code, Cursor) via MCP

• M.Sc. or Ph.D. in Computer Science, Machine Learning, or related field (or equivalent industry experience)

• 2+ years building production AI/ML systems (not just prototypes)
• Hands-on experience with LLM APIs (OpenAI, Anthropic) and prompt engineering
• Strong understanding of embeddings and vector databases (Weaviate, Chroma, pgvector)
• Proficiency in Python; working knowledge of TypeScript
• Track record of shipping AI features to end users

Preferred

• Experience with agentic LLM frameworks (LangChain, LlamaIndex, Pydantic AI, DSPy)
• Experience building LLM evaluation pipelines
• Familiarity with MCP (Model Context Protocol) or similar agent-tool interfaces
• Background in scientific/technical domains (physics, engineering, simulation)
• Production AWS experience (EC2, ECS, Lambda)
• Experience with containerization (Docker) and observability tooling
• Knowledge of traditional ML beyond LLMs

• Competitive compensation with equity of a fast-growing startup.
• Medical, dental, and vision health insurance.
• 401(k) Contribution.
• Gym allowance.
• Friendly, thoughtful, and intelligent coworkers.

As our market and products grow, we are rapidly expanding and searching for partners who are eager to grow in a dynamic environment, possess an entrepreneurial spirit, and can scale our team. Flexcompute is dedicated to providing equal employment opportunities. We firmly believe that talent from diverse backgrounds can bring our company a rich and varied perspective. We warmly welcome candidates from all backgrounds to join us on this passionate and challenging journey, together facing the most compelling challenges in engineering computation.

Flexcompute is dedicated to promoting diversity, equity, and inclusion in the workplace. We are an equal opportunity employer that recognizes the value of diverse perspectives in achieving our mission. We encourage candidates from all backgrounds to apply.