Machine Learning Software Engineer

Summary

Join PermitFlow, a company revolutionizing construction permitting, as a Machine Learning Engineer. You will play a key role in developing intelligent systems to enhance permit processing, document understanding, and compliance workflows. This involves working on LLM-based models, RAG pipelines, and AI-driven automation. You will design, implement, and optimize these models, working with large-scale data and collaborating with cross-functional teams. The ideal candidate possesses 3+ years of experience in machine learning engineering, strong understanding of LLMs and retrieval systems, and proficiency in Python and relevant ML libraries. PermitFlow offers competitive benefits, including equity packages, comprehensive health coverage, home office stipend, and more.

Requirements

• 3+ years of experience in machine learning engineering, ideally in production environments
• Strong understanding of LLMs (e.g., OpenAI GPT, Hugging Face models) and their applications
• Hands-on experience with retrieval systems (e.g., Elasticsearch, FAISS, or vector databases)
• Proficiency in Python and common ML libraries (e.g., TensorFlow, PyTorch, scikit-learn)
• Experience with ML model deployment, monitoring, and scaling in a cloud environment (AWS, GCP, or Azure)
• Strong problem-solving skills and ability to work in a fast-paced, high-ownership environment

Responsibilities

• Design, implement, and optimize LLM-powered models for document processing, data extraction, and permit application workflows
• Develop and fine-tune retrieval-augmented generation (RAG) pipelines to improve query processing and information retrieval
• Experiment with pre-trained models and fine-tune them for permit-related NLP tasks, such as document classification and entity recognition
• Build scalable machine learning infrastructure, integrating with backend systems to support AI-driven workflows
• Work with large-scale structured and unstructured data to ensure efficient indexing, retrieval, and contextual relevance
• Monitor and improve the performance and scalability of deployed models
• Stay updated with the latest research in LLMs, NLP, retrieval systems, and apply best practices to our AI models
• Collaborate with engineers, product managers, and legal experts to develop AI-native solutions for complex permitting challenges

Benefits

• Equity packages
• Competitive Salary
• 100% Paid health, dental & vision coverage
• Home office & equipment stipend
• Lunch & Dinner provided via UberEats w/ a fully stocked kitchen
• Commuter benefits
• Team building events
• Unlimited PTO