AI/ML Engineer

Softek Llc; Rockville (MD); On-site; USD 90,000 - 130,000; Full time

Job summary

Softek Llc in Rockville, Maryland, is seeking an AI/ML Engineer to develop and deploy machine learning models powering the Axiom multi-agent AI platform. In this role, you will focus on natural language processing and intelligent automation for federal agencies, processing millions of records.

The ideal candidate has over 3 years of experience in machine learning engineering, proficiency in Python, and a strong understanding of NLP techniques. A Secret clearance is required for this position.

Qualifications

• 3+ years of experience in machine learning engineering.
• Proficiency in Python and ML frameworks.
• Experience deploying ML models in production cloud environments.
• Strong understanding of NLP and document understanding techniques.
• Ability to obtain and maintain a Secret clearance.

Responsibilities

• Design, train, and deploy machine learning models for document processing and NLP tasks.
• Build and maintain ML pipelines for data ingestion, feature engineering, and model serving.
• Optimize model performance for accuracy, latency, and cost in production environments.
• Integrate AI capabilities into the Axiom platform architecture.
• Evaluate new ML techniques and frameworks for applicability to federal use cases.
• Document model architectures, training procedures, and performance benchmarks.

Skills

Machine Learning Engineering

Python

NLP Techniques

Cloud Environments

ML Frameworks (PyTorch, TensorFlow)

MLOps Tools

Tools

PyTorch

TensorFlow

scikit-learn

MLflow

Kubeflow

Job description

As an AI/ML Engineer at Softek, you will develop and deploy machine learning models that power our Axiom multi-agent AI platform. You will work on natural language processing, document understanding, and intelligent automation solutions used by federal agencies to process millions of records. You will collaborate with data scientists, engineers, and government stakeholders to turn research into production-ready AI systems.

Key Responsibilities

• Design, train, and deploy machine learning models for document processing and NLP tasks
• Build and maintain ML pipelines for data ingestion, feature engineering, and model serving
• Optimize model performance for accuracy, latency, and cost in production environments
• Integrate AI capabilities into the Axiom platform architecture
• Evaluate new ML techniques and frameworks for applicability to federal use cases
• Document model architectures, training procedures, and performance benchmarks

Required Qualifications

• 3+ years of experience in machine learning engineering
• Proficiency in Python and ML frameworks (PyTorch, TensorFlow, or scikit-learn)
• Experience deploying ML models in production cloud environments
• Strong understanding of NLP and document understanding techniques
• Ability to obtain and maintain a Secret clearance

Preferred Qualifications

• Experience with large language models and prompt engineering
• Familiarity with Azure AI services or AWS SageMaker
• Experience with MLOps tools (MLflow, Kubeflow, or similar)
• Prior work in federal or regulated environments

8(a) sole source up to $4.5M · HUBZone · GSA MAS