Staff AI Software Engineer
Hybrid · Toronto, Ontario, Canada
Job Summary
Staff AI Software Engineer responsible for owning end-to-end ML lifecycle for AI systems (from prototype to production) including NLP/LLM workloads, deployment on GPUs and Kubernetes, and production-grade engineering practices. Collaborates with product and business stakeholders, designs scalable architectures for AI/document intelligence features (Ask iManage, classification, extraction, NLP-driven capabilities), and mentors teammates. Emphasizes cost, performance, reliability, observability, CI/CD, model/version tracking, and governance. Requires strong Python/PyTorch expertise, hands-on experience with traditional ML/NLP and transformer architectures, and track record of deploying NLP/LLM solutions to production; bonus for knowledge graphs, multimodal LLMs, distributed training, and agentic tooling.
Required Qualifications
- Bachelor's, Master's or Ph.D. in Computer Science, Machine Learning, Data Science, Artificial Intelligence, Statistics, or a related field
- 4+ years of experience in ML/AI engineering or software engineering with 3+ years in NLP and LLM systems into production
- Proficiency in Python and PyTorch; experience with Hugging Face
- Hands-on experience with traditional ML/NLP and modern genAI/transformer architectures
- Experience fine-tuning language models and deploying them to production
- GPU optimization for training and inference
- Kubernetes deployment on cloud infrastructure (Azure/AWS/GCP)
- Ability to work across teams and own from prototype to production
- Strong communication and ownership skills
- Bonus: knowledge graphs, multimodal LLMs, distributed training, LangChain/LlamaIndex
Apply with one swipe on Sorce. We auto-fill applications and apply on your behalf — no cover letters, no 40-minute forms.
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.