Senior Software Engineer

About us:

Axiomatic AI is building a new class of AI systems designed to reason with the rigor of the scientific method. By combining deep learning with formal logic and physics-based modeling, we create verifiable, interpretable AI systems that collaborate with and support human researchers in high-stakes scientific and engineering workflows. 

Our mission, 30×30, is to deliver a 30× improvement in the speed, accessibility, and cost of semiconductor and photonic hardware development by 2030. 

We aim to revolutionize hardware design and simulation in these industries and are building a team of highly motivated professionals to bring these innovations from research into commercial products.

Position Overview

As Senior Software Engineer, you'll work fluidly across backend and frontend, shipping production-grade features end-to-end in an AI-native product. You won't be boxed into one side of the stack, you'll collaborate with both the backend and frontend teams, picking up whatever work has the highest impact and filling gaps wherever they appear.

You will:

• Ship features end-to-end: from database schema and API design through to pixel-perfect UI
• Collaborate across backend and frontend teams, contributing meaningfully to both codebases
• Design and build scalable backend services (FastAPI, Python) and responsive web applications (React, Next.js, TypeScript)
• Integrate AI capabilities (LLMs, agents) into production systems across the full stack
• Implement UX/UI patterns for AI-assisted scientific workflows when working on the frontend
• Ensure system reliability, performance, and security on both sides of the stack
• Mentor engineers and help establish cross-stack best practices as the team grows

We're looking for a versatile, T-shaped engineer someone with strong depth in either backend or frontend, and solid working proficiency across the full stack, who thrives on variety and can ship autonomously without waiting on hand-offs.

Key Responsibilities

1. Full-Stack Development

• Build and maintain backend services (FastAPI, Python) and web applications (React, Next.js, TypeScript)
• Design REST APIs, database schemas, and optimize SQL queries (PostgreSQL)
• Implement responsive, accessible, and performant UI components
• Integrate with external services (OpenAI, Anthropic, Google, Vertex AI) and surface AI features in the UI
• Debug across the entire stack: from SQL queries and network traces to React render cycles

2. Architecture & Design

• Contribute to system architecture and technical decisions across backend and frontend
• Design for scalability, reliability, and security in your areas of ownership
• Shape API contracts collaboratively — as both a producer and a consumer
• Write clear technical designs and propose improvements to the codebase and infrastructure

3. AI Integration

• Build and evolve agent flows using Pydantic AI / FastMCP
• Integrate LLM providers and design tool / function-calling patterns
• Design intuitive UX flows for AI-powered features (prompt editing, model selection, result visualization)
• Contribute to evaluation and regression testing for AI features

4. Quality & Testing

• Write comprehensive unit, integration, and end-to-end tests across the stack
• Participate in code reviews (give and receive feedback)
• Optimize performance on both backend (latency, throughput) and frontend (load times, bundle size)
• Debug and resolve production issues

5. Collaboration & Mentorship

• Work closely with the Tech Lead / Staff Engineer on architecture and roadmap
• Partner with AI Platform Engineer on AI integrations
• Bridge backend and frontend teams — translate needs and reduce hand-off friction
• Mentor engineers, share cross-stack knowledge, and participate in technical interviews

Key Requirements

• 6+ years of software engineering experience, with meaningful depth in at least one of backend or frontend
• Strong Python skills: FastAPI (preferred), Flask, or Django
• Strong JavaScript/TypeScript skills: React (Next.js preferred)
• Solid database fundamentals: PostgreSQL, SQL, ORMs (SQLAlchemy)
• REST API design experience: best practices, versioning, documentation
• Frontend fundamentals: CSS (Tailwind, SCSS), state management (TanStack Query, Zustand, Redux), responsive and accessible design
• Cloud platform experience: GCP preferred (AWS, Azure acceptable)
• Testing mindset: unit + integration + e2e tests, coverage discipline
• Version control & CI/CD: Git, GitHub Actions, Docker
• Strong problem-solving: debugging and performance optimization across the stack

Nice-to-Have

• FastMCP, Pydantic AI, or agent framework experiencAsync Python (asyncio, async/await) at scale
• Production AI/ML API integration (OpenAI, Anthropic, Gemini, Vertex AI)
• Experience building AI-powered UIs (chat interfaces, code editors, notebook-style interactions)
• Infrastructure as code (Terraform)
• Microservices and event-driven architectures
• Knowledge of WebSockets, SSE, and real-time communication
• Observability tools (Sentry, Logfire, OpenTelemetry)
• Figma or similar design tool proficiency
• Background in scientific or technical product UX

Current Stack:

• Backend: Python, FastAPI, SQLAlchemy, Pydantic AI, FastMCP, Alembic
• Frontend: React (Next.js 15), TypeScript, App Router, Tailwind CSS, shadcn/ui, Zustand, TanStack React Query
• Databases: PostgreSQL, Redis (caching)
• APIs: REST, WebSockets, SSE, MCP
• AI/ML: Anthropic Claude, Google Gemini, OpenAI, Vertex AI Model Garden, Mistral OCR, CopilotKit, LangGraph
• Cloud: Google Cloud Platform (Cloud Run, Cloud SQL, GCS, VPCs, Auth0)
• Infrastructure: Terraform, Docker
• CI/CD: GitHub Actions
• Observability: Logfire, Sentry
• Testing: pytest, pytest-asyncio, pytest-cov, Jest, Playwright, Cypress
• Design: Figma

Work model & location expectations:

• Team work model:
  Preferred hybrid from our Boston office; remote arrangement may be considered.
  
  
• Primary location:
  Boston, US
  

Why join us? 

At Axiomatic_AI, you will be working on technology that drives innovation in AI for scientific and engineering applications in line with our 30X30 mission. 

This is your opportunity to contribute to the development of new AI architectures that can reason coherently and produce interpretable and verifiable solutions. Consequently, see those ideas commercialized into products that will shape the future of hardware and computing, while collaborating with a global team of engineers and AI specialists. 

We believe in pushing the boundaries of what is possible and continuously seek to redefine the intersection of AI, with focus on formal consistency. If you're ready to take your expertise in artificial intelligence and physics to the next level, we want to hear from you!

Worried about not meeting every qualification? Studies show that women and people of color are less likely to apply for jobs unless they meet every listed requirement. At Axiomatic-AI, we are dedicated to creating a diverse, inclusive, and authentic workplace. If this role excites you but your background doesn’t perfectly match every qualification, we still encourage you to apply. You could be the perfect fit for this position or another opportunity with us.