Skip to content
WebAI

Senior Machine Learning Engineer

Austin, TX; Washington D.C. AreaUnited States
Senior · FullTime · Public Sector
POSTED 16 APRVERIFIED 9H AGO
Apply Ask
Pay · H-1B filings
$150k
median · n=6 · FY26
Open for
200d
since 16 Apr
H-1B · FY26
9
certified LCAs
Hiring · 30d
19
0 net · open roles
PyTorchTensorFlowLLMsDistributed systems

Special Notice

This position is NOT contingent upon awarding of a project or needing a funding source. This is full-time employment with webAI.

About the Role

We are seeking a Senior Machine Learning Engineer to support our Public Sector initiatives focused on building and optimizing production ready AI systems for secure and distributed environments.

You will be responsible for transforming prototype models into scalable, efficient, and reliable production systems that operate seamlessly across a spectrum of hardware from government cloud infrastructure to edge devices in restricted or disconnected environments.

Responsibilities

Design, develop, and deploy agentic workflows to orchestrate multi-step reasoning, tool use, and decision-making across production systems.

Productionize AI models from research prototypes into scalable, deployable systems used in real world applications.

Engineer adaptive ML systems using LoRA, PEFT, and on-device inference strategies, leveraging PyTorch, TensorFlow, and Hugging Face Transformers for model development, fine-tuning, and optimization.

Implement model optimization techniques such as quantization, pruning, distillation, and hardware specific acceleration.

Build and maintain Retrieval Augmented Generation (RAG) pipelines, including vector database integration for contextual retrieval.

Work with multi-modal AI systems across computer vision, audio, and natural language domains.

Optimize model execution for distributed and resource constrained environments, ensuring reliability under variable connectivity conditions.

Qualifications

Active US Security clearance

4+ years of experience in applied AI, ML engineering, or production AI systems.

Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers.

Proven experience deploying AI models across cloud, edge, and mobile hardware environments.

Expertise in model compression and optimization (quantization, pruning, distillation).

Experience building RAG pipelines and integrating vector databases (e.g., Quadrant, ChromaDB, FAISS, Milvus, Pinecone).

Familiarity with multi-modal models and synthetic data generation methods.

Strong algorithmic and problem solving skills, especially in distributed or constrained compute environments.

Preferred Skills

Experience with edge AI, federated learning, or offline inference systems.

Understanding of AI governance and compliance frameworks relevant to public sector deployments.

Experience integrating models into large scale distributed systems or microservice architectures.

Excellent communication and technical documentation skills for collaboration across multi disciplinary teams.

Strong understanding of GPU computing, CUDA, and performance profiling.

We at webAI are committed to living out the core values we have put in place as the foundation on which we operate as a team. We seek individuals who exemplify the following:

Truth - Emphasizing transparency and honesty in every interaction and decision.

Ownership - Taking full responsibility for one’s actions and decisions, demonstrating commitment to the success of our clients.

Tenacity - Persisting in the face of challenges and setbacks, continually striving for excellence and improvement.

Humility - Maintaining a respectful and learning-oriented mindset, acknowledging the strengths and contributions of others.

Benefits

We strive to provide competitive benefits to all employees. The benefits listed in this posting generally apply to U.S.-based employees. For employees hired outside the United States, benefits may vary based on local law, country-specific requirements, and the employment platform or entity through which the employee is hired.

Competitive salary and performance-based incentives

Comprehensive health, dental, and vision benefits package

Including a high-deductible HSA option with company contribution

401k match (US-based only)

Flexible PTO Policy

Parental leave for eligible employees

Continuing education support

Basic, company-paid life insurance

Supplemental life insurance

Company-paid short and long term disability

Source · WebAI · AshbyFirst seen 2 NovOriginal ↗Method