webAI

Senior Machine Learning Engineer

webAI
Apply
7 months ago
Washington, DC, USA or Austin, TX, USASenior
H1B sponsor

Responsibilities

  • Design, develop, and deploy agentic workflows for multi-step reasoning, tool use, and production decision-making systems.
  • Productionize AI models from research prototypes into scalable, deployable real-world systems.
  • Develop and optimize adaptive ML systems using LoRA, PEFT, PyTorch, TensorFlow, and Hugging Face Transformers.
  • Apply quantization, pruning, distillation, and hardware-specific acceleration techniques.
  • Build and maintain RAG pipelines with vector database integrations for contextual retrieval.
  • Develop multimodal AI systems across computer vision, audio, and natural language domains.
  • Optimize model execution across distributed and resource-constrained environments with variable connectivity.

Requirements

  • Active US security clearance is required.
  • At least 4 years of experience in applied AI, ML engineering, or production AI systems is required.
  • Deep proficiency in PyTorch, TensorFlow, or Hugging Face Transformers is required.
  • Experience deploying AI models across cloud, edge, and mobile hardware is required.
  • Expertise in model compression and optimization, including quantization, pruning, and distillation, is required.
  • Experience building RAG pipelines and integrating vector databases such as Quadrant, ChromaDB, FAISS, Milvus, or Pinecone is required.
  • Familiarity with multimodal models and synthetic data generation methods is required.
  • Strong algorithmic and problem-solving skills in distributed or constrained compute environments are required.
  • Preferred qualifications include experience with edge AI, federated learning, or offline inference systems; public-sector AI governance and compliance; large-scale distributed or microservice architectures; GPU computing, CUDA, and performance profiling; and strong communication and technical documentation skills.

Benefits

  • Comprehensive health, dental, and vision benefits, including a high-deductible HSA option with company contribution.
  • 401(k) match for US-based employees.
  • Flexible paid time off policy.
  • Parental leave for eligible employees.
  • Continuing education support.
  • Basic company-paid life insurance and supplemental life insurance.
  • Company-paid short- and long-term disability coverage.
  • Free parking for in-office employees.
  • Benefits may vary for employees hired outside the United States based on local law, country-specific requirements, and hiring entity.

Tech Stack

Hugging Face TransformersPyTorchTensorFlow

Categories

webAI

About webAI

51-200 employees

webAI builds an end-to-end platform for training, deploying, and operating private AI models on infrastructure customers control, including edge devices. It sells enterprise software and developer SDKs/APIs used by public-sector, defense, aviation, and manufacturing teams to run specialized models with data sovereignty. Founded in 2020 and headquartered in Austin, Texas, the privately held company focuses on distributed AI/LLM infrastructure across local and cloud environments.

Contact me