What are the responsibilities and job description for the Python Engineer (AI Inference) position at Mathpix?
About Mathpix:
We are a small (but quickly-growing) and dedicated team committed to pushing the boundaries of what is possible in computer vision and AI-powered document processing. We are profitable and help power some of the world's biggest LLMs providers, EdTech, and AI companies. We are headquartered in Williamsburg, Brooklyn, NY but have a small remote team as well.
About the role:
We are seeking a Python engineer to join our team in the Brooklyn office, working on the AI inference pipeline. This powers our industry-leading OCR and computer vision products. We are open to junior and more experienced candidates for this position!
You’ll work closely with senior ML engineers to add new OCR features, improve existing models, and build tools that make experimentation and deployment faster and more reliable. It’s is an exciting opportunity for an early-career developer who is passionate about algorithms, developer productivity, and machine learning, and who wants to contribute directly to advancing OCR technology.
Responsibilities:
- Collaborate with senior ML engineers on developing and integrating new OCR features into our inference pipeline
- Write clean, object-oriented Python code that supports scalability and maintainability
- Work with SQL databases for data access, storage, and analysis in support of ML training and evaluation
- Contribute to model development and evaluation for real-world computer vision problems
- Implement algorithms and data structures in production code
- Help build developer tools and utilities that make our ML workflows more efficient and reproducible
Required skills:
- Proficiency in Python, with a solid understanding of object-oriented programming
- Experience implementing algorithms (e.g., depth-first search, recursive descent parsers)
- Experience interacting with SQL databases
- Strong interest in machine learning, computer vision, and developer tooling
- Desire to learn from and collaborate with experienced engineers on real-world ML systems
Additional desired experience:
- PyTorch experience
- Familiarity with containerization (Docker, Kubernetes) or cloud platforms (AWS, GCP, Azure)
- Previous contributions to open-source projects, coursework, or personal projects related to ML or developer tooling
Mathpix values:
- A strong sense of ownership and accountability for the systems you build
- A collaborative mindset with a willingness to mentor and be mentored
- Passion for solving complex problems with elegant, pragmatic solutions
- Resilience and self-direction — the ability to thrive in a fast-paced startup environment, where priorities evolve and autonomy is key