AI Engineer

• Location: Palo Alto or Remote
• Company Stage of Funding: Seed
• Office Type: On-site in Palo Alto or Remote
• Salary: $300K - $400K + 1-5% Equity

Company Description

Our client is a cutting-edge LegalTech startup revolutionizing the immigration industry through innovative AI solutions. Backed by prestigious venture capital firms, they combine legal expertise with advanced technology to streamline complex immigration processes. Their comprehensive platform empowers individuals and businesses to navigate immigration challenges with confidence and ease.

What You Will Do

• Collaborate closely with the founding team and early customers to build and ship AI-native products
• Transform complex AI research into actionable solutions that deliver measurable business value
• Design and implement LLM-powered products, including prompting, agents, experimentation, LLM Ops, and performance measurement
• Evaluate and productize various types of LLMs for specific product use cases
• Research and stay current with the latest advancements in NLP such as advanced prompting, function calling, embeddings, and building agents
• Own your work end-to-end and integrate research findings into product development

Ideal Candidate Background

• 3+ years of experience in engineering with AI/ML, LLM, and evaluation techniques
• Experience with LLM operations including pre-training, fine-tuning, and inference optimization
• Track record of successfully building and deploying LLM-based applications
• Experience in AI/ML roles at impactful research teams or leading technology companies
• MS/PhD in Computer Science, Artificial Intelligence, Machine Learning, or related fields
• Hands-on experience with LLM Ops, prompt engineering, experimentation, and performance evaluation
• Proficiency in Python with practical experience in PyTorch, TensorFlow, and Hugging Face Transformers

Preferred Qualifications

• Experience working in startup environments
• Background in applied AI/ML research with publications in recognized conferences
• Experience with distributed training of LLMs using cloud platforms or HPC environments
• Advanced NLP techniques experience, particularly in multilingual capabilities and contextual understanding
• Product sense and ability to evaluate various LLM models for specific use cases

Compensation and Benefits

• Competitive salary range: $300K - $400K
• Generous equity package: 1-5%
• Visa sponsorship available for qualified candidates
• Flexible work arrangements: on-site in Palo Alto or fully remote
• Opportunity to join a seed-stage company with significant growth potential
• Work directly with accomplished founders from top companies and institutions