Pytorch Developer Jobs in Virginia (NOW HIRING)

We are looking for a skilled AI Developer to design, build, and optimize advanced AI solutions across predictive, generative, and autonomous system domains. This role requires solid hands-on engineering capabilities, familiarity with modern AI architectures, and the ability to translate mission needs into robust, production-ready AI capabilities. The AI Developer will work across the full stack of AI development, from data ingestion and model experimentation to application integration, orchestration, and deployment, and will collaborate closely with product teams, LLMOps engineers, designers, and mission stakeholders. 

• Develop end-to-end AI solutions including LLM-powered applications, predictive ML models, multi-agent workflows, RAG pipelines, and specialized AI microservices. 

• Implement reusable AI components, libraries, and APIs that streamline application development and accelerate delivery across programs. 

• Integrate AI models with enterprise systems, APIs, data platforms, vector databases, and cloud-native services to deliver scalable mission capabilities. 

• Drive iterative experimentation, prototyping, and model improvement cycles in collaboration with Data Scientists and AI Evaluation Scientists. 

• Design and implement advanced prompt strategies, context management layers, retrieval systems, and LLM orchestration logic. 

• Build scalable inference services, optimize model performance, and collaborate with LLMOps to enable robust deployment, monitoring, and continuous improvement. 

• Translate user needs and mission workflows into intuitive, reliable AI-powered features through active partnership with designers and product teams. 

• Implement secure-by-design and trustworthy AI practices, including safety guardrails, input sanitization, content filtering, and integration of evaluation metrics. 

• Contribute to internal AI frameworks, code patterns, and shared accelerators that raise delivery quality across the AI & Data Exploitation Practice. 

• Participate in code reviews and support engineering excellence across multi-disciplinary AI delivery teams. 

• Stay current with emerging AI techniques, libraries, foundation models, and agent frameworks, evaluating their applicability to client missions. 

• You will contribute to the growth of our AI & Data Exploitation Practice! 

• Ability to hold a position of public trust with the U.S. government. 

• Bachelor's, Master's, or Ph.D. in Computer Science, Artificial Intelligence, Machine Learning, Software Engineering, or a related field. 

• 2+ years of hands-on software engineering experience, with exposure to AI/ML, generative AI, or LLM-driven application development. 

• Strong proficiency in Python and modern AI frameworks such as PyTorch, TensorFlow, Hugging Face Transformers, LangChain, LlamaIndex, or similar. 

• Demonstrated ability to design and develop production-grade AI applications, including APIs, back-end services, orchestration logic, and front-end integrations (when needed). 

• Experience implementing RAG architectures, embeddings, vector stores, and context retrieval patterns. 

• Familiarity with multi-agent orchestration frameworks, prompt engineering strategies, and advanced LLM interaction design. 

• Strong understanding of cloud platforms (AWS, Azure, GCP), including compute, serverless services, and security fundamentals for AI workloads. 

• Working knowledge of containerization (Docker), orchestration (Kubernetes), and CI/CD pipelines for AI-based systems. 

• Experience with structured and unstructured data, document processing, and application integration with existing enterprise systems. 

• Understanding of responsible AI principles including safety, fairness, privacy, and model risk mitigation. 

• Strong analytical and communication skills with the ability to collaborate across engineering, design, and mission domains. 

• A collaborative mindset and a commitment to raising the technical quality of development work. 

Steampunk relies on several factors to determine salary, including but not limited to geographic location, contractual requirements, education, knowledge, skills, competencies, and experience. The projected compensation range for this position is $115,000 to $145,000.  The estimate displayed represents a typical annual salary range for this position. Annual salary is just one aspect of Steampunk's total compensation package for employees. Learn more about additional Steampunk benefits here. 

Identity Statement

As part of the application process, you are expected to be on camera during interviews and assessments. We reserve the right to take your picture to verify your identity and prevent fraud.

Steampunk is a Change Agent in the Federal contracting industry, bringing new thinking to clients in the Homeland, Federal Civilian, Health and DoD sectors. Through our Human-Centered delivery methodology, we are fundamentally changing the expectations our Federal clients have for true shared accountability in solving their toughest mission challenges. As an employee owned company, we focus on investing in our employees to enable them to do the greatest work of their careers - and rewarding them for outstanding contributions to our growth. If you want to learn more about our story, visit http://www.steampunk.com. 

We are an equal opportunity employer and all qualified applicants will receive consideration for employment without regard to race, color, religion, sex, national origin, disability status, protected veteran status, or any other characteristic protected by law.Steampunk participates in the E-Verify program.