Lead Software Engineer, AI

- 5+ years in applied ML/AI or 5+ years with an advanced degree, delivering production systems in document AI/OCR/NLP or information extraction at scale. - Strong expertise in Python and deep learning (PyTorch/TensorFlow), with knowledge in LLMs, embeddings/vector search, prompt design/evaluation, and statistical methods for quality and uncertainty. - Experience with cloud-native ML operations: training/inference pipelines, feature/data storage, observability, cost/performance optimization. - Ability to manage the end-to-end lifecycle for AI features: design, implementation, testing, deployment, monitoring, and iteration. - Proficiency in optimizing model/workflow inference for latency, throughput, and cost, including strategies for caching, batching, and model selection. - Experience in implementing monitoring, logging, tracing, and alerting for workflows and models in production to track data drift, model performance, and error rates. - Proven ability to debug and troubleshoot complex AI systems in production, including prompt failures, integration bugs, and data quality issues. - Capability to collaborate with Product, Tax Subject Matter Experts, and other engineering teams to translate business and tax requirements into AI workflows. - Experience working alongside data scientists and ML engineers on model selection, evaluation, and experimentation within pipelines. - Demonstrated leadership in conducting design and code reviews while mentoring mid-level/junior engineers on AI best practices and production-level coding. - Contribution to internal standards and best practices for developing AI workflows with a focus on effective documentation and templates. - Experience in creating end-to-end AI features that are deployed in production environments, not merely prototypes. - Ability to convert vague tax/business requirements into clear, testable workflow designs. - Commitment to robustness and accuracy, especially in areas where mistakes can have significant financial or compliance implications. - Willingness to engage with tax/business regulations and collaborate with domain experts, even without prior tax specialization.