AI/ML Engineer I

Torch.AI builds government-owned reasoning infrastructure that enables machine reasoning at scale for defense, national security, and operational environments. The AI/ML Engineer I will implement and evaluate machine learning components, support data and NLP pipelines, build AI services, and learn deployment and model lifecycle practices under senior guidance.

What you'll do:

• Implement and evaluate machine learning components under senior guidance, focusing on operational performance rather than academic novelty
• Support data preparation, labeling, preprocessing, and feature engineering across structured and unstructured datasets
• Contribute to retrieval, embedding, or LLM-enabled pipelines with mentorship
• Assist in building prototype and production-ready AI services using frameworks such as PyTorch, scikit-learn, or modern NLP toolkits
• Execute experiments for NLP/NLU workflows including classification, entity extraction, semantic search, and retrieval
• Support model evaluation using measurable metrics aligned with latency, accuracy, and mission constraints
• Learn foundational deployment patterns including containerization, API exposure, monitoring, and model lifecycle practices
• Participate in code reviews, sprint planning, documentation, and disciplined software development practices

What they're looking for:

• * B.S. in Computer Science, Engineering, Mathematics, or related technical field
• * 1–3 years of experience in software engineering or applied ML (internships included)
• * Proficiency in python
• * Strong Python fundamentals and ability to write clean, testable code
• * Familiarity with at least one ML/NLP library (e.g., scikit-learn, spaCy, PyTorch, TensorFlow, Transformers)
• * Foundational understanding of ML workflows: data preprocessing, training loops, validation, and evaluation metrics
• * Ability to work in ambiguous problem spaces and iterate quickly with guidance
• * Strong communication skills and willingness to collaborate across engineering disciplines
• U.S. citizenship is required for all positions. Torch.AI does not sponsor employment visas
• * Limited travel (<10%) may be required for some roles
• * Exposure to LLMs, embeddings, semantic search, or retrieval pipelines
• * Familiarity with vector databases and/or graph-based data systems

Benefits:

• Quarterly performance bonuses
• Equity participation within the first 12 months
• Unlimited PTO + 11 paid company holidays
• Professional development in a high-growth, mission-driven environment
• Weekly in-office catering at HQ
• 401(k) plan (no current employer match, but under consideration)
• PPO, HSA, and TRICARE Supplement medical options
• Above-market HSA contributions
• HSA, FSA, and Dependent Care FSA options
• Dental and vision plans above national averages