Applied AI Engineer
Job Description
Applied AI Engineer role at Soulside AI focused on owning the model layer for trustworthy behavioral health documentation, reporting to the CTO.
Responsibilities
- Build post-training pipelines for open-source models, including SFT, DPO/RLHF preference optimization, LoRA/adapters, and distillation for domain-specific clinical tasks.
- Fine-tune, deploy, and serve models across managed inference and fine-tuning platforms such as Fireworks AI, Baseten, and Together AI; make build-vs-buy decisions to match each workload.
- Design and maintain rigorous evaluation sets for high-stakes outcomes like clinical reasoning and AI note generation, including metrics, gold-standard data curation, and automated plus human-in-the-loop eval harnesses.
- Implement a fast iteration loop from eval results to training changes and releases, identifying regressions early and quantifying impact from model or prompt updates.
- Optimize the complete LLM pipeline: prompting, retrieval, structured output validation, plus latency and cost.
- Work with clinical experts to translate documentation and compliance requirements into model behavior and evaluation criteria.
- Monitor models in production for quality, drift, and failure modes, then feed insights back into training data and evaluation sets.
Requirements
- 3+ years in applied ML/AI engineering, or a Master’s degree in a related field, with hands-on experience bringing LLM-based systems into production.
- Practical experience with post-training and fine-tuning open-source models (e.g., Llama, Qwen, Mistral) using SFT, LoRA/PEFT, or preference-based methods.
- Experience serving and/or fine-tuning on managed platforms such as Fireworks AI, Baseten, or Together AI (or similar inference/training infrastructure).
- Proven ability to build evaluation frameworks for LLM tasks, focusing on measurable quality.
- Python proficiency and familiarity with modern ML tooling (including PyTorch and Hugging Face).
- Strong grounding in prompt engineering and structured-output validation.
- Ability to thrive in a fast-paced, remote-first startup environment and communicate clearly with technical and clinical stakeholders.
- Visa sponsorship available for the right candidate, including H-1B and O-1.
Technologies
- Python, PyTorch, Hugging Face
- Fireworks AI, Baseten, Together AI
- Llama, Qwen, Mistral
- SFT, LoRA, PEFT, DPO, RLHF, LoRA/adapters, distillation
- Retrieval, structured output validation
Benefits
- Salary: USD 150,000–200,000 per year, plus equity with significant upside potential as a founding team member.
- Comprehensive health, dental, and vision insurance.
- Flexible, remote-first culture.
- Direct access to founders and influence on technical direction.
- Professional development budget and conference attendance.
- Opportunity to build AI that measurably improves mental health care at scale.
Bonus Points
- Experience with healthcare, clinical NLP, or other high-stakes regulated domains.
- Familiarity with HIPAA and handling sensitive clinical data.
- RAG systems, retrieval quality tuning, or long-context document workflows.
- Experience with LLM observability, monitoring, and drift detection in production.
- Experience with data pipelines and labeling workflows for curating high-quality training and eval sets.
- Open-source contributions in the ML/LLM ecosystem.
How to Apply
- Send your resume and a short note to [email protected].
- Include a model or pipeline you took to production and explain how you validated it was actually working.