AI/ML Engineer
About The Position
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. Role Overview 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.
Requirements
- 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
Nice To Haves
- 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
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
Benefits
- 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.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
Number of Employees
51-100 employees
