Software Engineer, AI

Document Crunch is seeking a mid-level Software Engineer, AI to join our amazing, tight-knit team.

We are looking for a Software Engineer, AI with expertise in Retrieval-Augmented Generation (RAG), AI/ML research, model training/fine-tuning, and ML service(s) integration. The ideal candidate will have a demonstrated history in machine learning, machine learning pipelines / infrastructure, natural language processing, prompt engineering, software engineering and some general devops / infrastructure. You will be responsible for researching, designing, developing, and deploying AI/ML services, and integrating them into our existing systems to enhance the functionality and efficiency of our products.

What makes  a great place to work? 

We’re cutting-edge innovators. We’re changing an industry. We’re making a real difference. Document Crunch builds AI-powered software tools for the built world. As the construction industry’s only document compliance platform, we simplify complex construction contracts and project documents, identifying critical risk provisions and providing teams with real-time guidance to make great decisions and comply with contractual obligations throughout the project lifecycle. We empower the industry, make it more efficient, reduce risks, and protect profits. Our company is people-focused and values-driven. We know that to achieve great things, you need a great team, and our “CRUNCHERS” go above and beyond in everything they do. In addition, everyone on our team stays true to our core values, a must if you’d like to become a Cruncher. 

We Are Warmhearted

Our success is powered entirely by our people who are real and approachable. We foster an environment that is warm, fun and welcoming for everyone. 

We Are Inspired

We care deeply about our mission and are inspired by our potential. If it was easy, everybody would do it, but we are striving for greatness. We take pride in being owners in this journey.

We Are Growth-Minded

Life is short and we want to make our time count. We invest in growth—personally and professionally. We never stop chasing our why.

Key Responsibilities

Lead AI/ML Strategy and Implementation 

• Understand and execute the company's AI/ML roadmap, aligning with business objectives 

• Design, implement and deploy advanced RAG systems to enhance document understanding and analysis capabilities 

• Lead the integration of cutting-edge AI/ML solutions into existing products and services

• Work closely with DevOps engineers and cross functional teams to integrate AI/ML solutions into existing workflows or be able to design and deploy new workflows as needed 

Advanced AI/ML Research and Development 

• Spearhead research initiatives on state-of-the-art AI/ML algorithms and technologies 

• Design and conduct experiments to solve challenging problems in document analysis and automation 

• Work independently to ideate, research, prototype and experiment with novel RAG stack solutions 

ML Architecture and Systems Design 

• Architect scalable, efficient, and maintainable ML systems and pipelines 

• Optimize ML system integration performance and resource utilization 

• Design and implement ML monitoring and observability systems 

Team Leadership and Mentorship 

• Serve as an experienced source of ML knowledge to less experienced or non-technical cross functional team members 

• Foster a culture of innovation, continuous learning, and best practices within the team 

• Collaborate with cross-functional leaders to align AI initiatives with product and business strategies 

Continuous Improvement and Innovation 

• Stay at the forefront of AI/ML advancements and industry trends 

• Identify and evaluate new technologies and methodologies for potential adoption 

• Drive the adoption of best practices in AI systems evaluation and benchmarking 

Education 

• Bachelor’s in Computer Science, Machine Learning, or a related field 

Technical Skills 

• Proficiency in Python 

• Strong understanding of NLP algorithms, including transformer architectures and large language models 

• Extensive experience with RAG systems, knowledge graphs, vector databases, and semantic search technologies 

• Advanced knowledge of cloud platforms (AWS, Azure, Google Cloud) and their ML services 

• Expertise in LLM prompt engineering, fine-tuning, and evaluation 

Experience 

• Multiple years of professional experience in AI/ML related engineering roles 

• Proven track record of leading complex AI/ML projects from conception to production 

• Extensive experience with model deployment, MLOps, and maintaining ML systems in production, or equivalent experience with ML systems integration at scale 

• Demonstrated ability to translate business requirements into technical solutions 

Leadership and Soft Skills 

• Strong leadership and team management skills 

• Excellent problem-solving and analytical abilities 

• Outstanding communication skills, able to explain complex technical concepts to diverse audiences 

• Proven ability to influence and drive decision-making across teams and departments 

Preferred Qualifications

• Experience in the construction or legal tech industry 

• Contributions to open-source ML projects or research publications 

• Experience in AI/ML research and development, with a focus on NLP and RAG at scale 

• Hands-on experience leading the integration of ML models into production environments 

We are currently hiring in Austin, TX and Atlanta, GA. While we are a remote first workplace, we want Crunchers in these locations to better facilitate periodic in-person meetings, get-togethers and training.

• Competitive salary and benefits package.
• Close to the ground floor of an incredibly high-growth business
• Substantial internal growth opportunities and emphasis on personal & professional development
• Generous paid time off policy
• World class medical, dental, and vision benefits
• 401k
• Participation in the employee stock option program
• Open-minded and collaborative work environment with a focus on results

We value diversity, equity, and inclusion in everything we do. It is our policy to provide equal opportunity to all people without regard to race, color, religion, national origin, marital status, veteran status, age, disability, pregnancy, sex, sexual orientation, gender identity or any other legally protected category.