Principal AI Researcher
Actively Hiring
Full-time $250k - $325k Posted 6 months ago
Responsibilities
- check_circle Design, build and scale algorithms for content understanding, inference, entailment, relationship mining, retrieval, etc. in production-grade attribution stack.
- check_circle Post-train foundation models (LLMs, VLMs) for content understanding and attribution solutions. Prototype, validate, and iterate rapidly on solutions to problems related to content understanding and source attribution.
- check_circle Develop efficient extraction and indexing pipelines for deep understanding and representation of content for entailment and source verification in AI generated answers and other NLU scenarios.
- check_circle Contribute to optimizing performance and latency for both real-time and batch use cases.
- check_circle Support identifying and addressing bottlenecks in the attribution stack to improve response times and system efficiency.
- check_circle Collaborate with cross-functional teams to integrate LLMs and attribution components into B2B and B2C services / applications.
- check_circle Evaluate and integrate modern frameworks and tools to accelerate development.
- check_circle Guide architectural decisions, mentor team members, and uphold engineering excellence.
- check_circle Author technical papers or patents.
- check_circle PhD degree in AI or related field, or equivalent practical experience.
- check_circle 8+ years of experience in software engineering, with a focus on NLP/NLU, Symbolic and/or neural networks & reasoning topics, AI/ML and distributed systems.
- check_circle Hands-on experience building and deploying content understanding and inference systems.
- check_circle Experience working on high reliability, observability, and monitoring of deployed systems is a plus
- check_circle Deep knowledge of Entity, Event, Proposition extraction, NLI, KGs.
- check_circle Experience with Search (OpenSearch, etc.) & Graph (Neo4J, etc.) databases.
- check_circle Strong coding skills in Python and/or other backend languages (e.g., Rust, Java).
- check_circle Experience with vector search, embedding pipelines, and dense retrieval techniques.
- check_circle Excellent problem-solving, analytical, and debugging skills.
- check_circle Strong sense of ownership, ability to work independently, and a self-starter mindset in fast-paced environments.
- check_circle Passion for building impactful technology aligned with our mission.
Preferred qualifications
- Proficiency in PyTorch, JAX, or other deep learning framework.
- Experience with various LLM post-training techniques.
- Significant experience with semantic search, embedding models, and prompt engineering.
- Publications at top-tier AI conferences (e.g., NeurIPS, ICML, CVPR, ICLR).
- Experience with proof, justification or compliance use cases.
- Previous work on systems handling millions of users or queries per day.
- Familiarity with cloud infrastructure (AWS, GCP, or Azure) and containerization tools (Docker, Kubernetes).
Benefits
- check_circle Opportunity to work at the forefront of AI technology
- check_circle Collaborative and innovative work environment
- check_circle Competitive salary and benefits package
- check_circle Professional development and growth opportunities
- check_circle Chance to make a significant impact on the company's success
- check_circle ProRata is an Equal Opportunity Employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. All employment decisions are made based on qualifications, merit, and business needs.
- check_circle At-Will Employment: Employment at ProrataAI is at-will. This means that either the employee or the employer may terminate employment at any time, with or without cause or prior notice.
- check_circle Salary Disclosure: In compliance with California law, salary information is provided to ensure transparency and fairness.
- check_circle California Consumer Privacy Act (CCPA): ProrataAI complies with the CCPA. Personal information collected during the recruitment process will be used for employment purposes only.
- check_circle This open position is not eligible for Company sponsorship of a visa that would require a new H-1B visa petition that is subject to the $100,000 payment requirement announced in the Presidential Proclamation titled “Restriction on Entry of Certain Nonimmigrant Workers,” dated September 19, 2025 (or any extensions or modifications of the Proclamation).
Tags & Focus Areas
Fulltime Ai Machine Learning Nlp