Senior AI/ML Engineer
Core
Build and test model and software improvements for the U.S. Census Bureau's LLM Autocoder to assign standardized codes to survey write-in responses.
Role type
Senior AI/ML Engineer (NLP/LLM)
Builds
Production ML models and software components for the American Community Survey (ACS) and other federal surveys
Domain
Federal government / Census data processing
Deliverable
production ML models
Required skills
Python, NLP, LLM integration, text classification, prompt design, automated testing, AWS, Git, data processing
Preferred skills
fuzzy matching, dictionary-based coding, survey text classification, ML frameworks (scikit-learn, Pytorch, Hugging Face), handling sensitive data in controlled environments
Technologies
Python, AWS, Git, scikit-learn, PyTorch, Hugging Face
Responsibilities
Develop Python components for text preprocessing and model invocation; Implement changes to prompts and model configurations; Build pilots using LLMs and hybrid approaches; Write comprehensive unit and integration tests; Reproduce and resolve production issues; Partner with MLOps on deployment and rollback; Maintain technical documentation
Seniority
Senior, hands-on IC