SquarePeg.ai is looking for an AI/NLP Engineer to serve as the technical backbone for our core matching algorithms. As a key force on our data and machine learning team, you will enhance AI systems that intelligently connect candidates with jobs and own the end-to-end machine learning lifecycle for a fast-growing HR Tech startup.

What You'll Do

Build and maintain taxonomies for candidate and job attributes; bootstrap gold datasets and evaluation pipelines.
Extract and normalize entities from resumes and job descriptions; craft and optimize prompts and fine-tuned models.
Develop and refine retrieval, ranking, and scoring using embedding-based methods and LLMs.
Refine proprietary scoring algorithms that evaluate candidate-job compatibility.
Conduct deep-dive analyses to identify patterns in successful hires and optimize the recommendation engine.
Implement innovative NLP solutions that understand context, intent, and nuance in hiring language.
Design and implement robust data pipelines that can handle massive volumes of resume and job posting data.
Build sophisticated entity resolution systems to normalize and deduplicate candidate profiles across multiple data sources.
Create scalable data architectures that power real-time matching at scale.
Collaborate directly with the product team to translate business requirements into technical solutions.
Own the end-to-end ML lifecycle from experimentation to production deployment.
Continuously iterate on algorithms based on customer feedback and performance metrics.

What We're Looking For

Deep understanding of machine learning algorithms, particularly in recommendation systems or ranking problems.
Experience with prompt engineering, prompt chaining, and LLM fine-tuning.
Knowledge of vector databases and semantic search technologies.
Familiarity with A/B testing and experimental design.
3+ years of hands-on experience with Python, SQL, and modern ML frameworks (PyTorch, TensorFlow, scikit-learn).
Proven track record in NLP and working with large language models (OpenAI, Anthropic, open-source LLMs).
Experience with data engineering tools and cloud platforms (AWS, GCP).
Strong background in entity resolution, data matching, or similar deduplication challenges.
Experience building and maintaining ontologies.
Experience building datasets and evaluation pipelines.
Ability to choose different methods based on tradeoffs of cost, latency, and accuracy.
Opinionated, data driven, and intellectually curious.
Ability to thrive in an environment where you experiment and move quickly.
Strong sense of ownership; can work autonomously.

Technical Stack

Languages: Python, SQL
ML Frameworks: PyTorch, TensorFlow, scikit-learn
LLMs: OpenAI, Anthropic
Cloud: AWS, GCP
Data: Vector databases

Team & Environment

You'll be joining a founding team of 12 and will work directly with the founding team on a product solving a real market problem. This is a fast-growing HR Tech startup where you will have a direct impact on the product and company success in a cutting-edge tech environment.