Senior Machine Learning Engineer
webAI- Location
- Austin, TX
- Workplace
- —
- Employment
- Full Time
- Salary
- —
Posted 5mo ago
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
- Comprehensive health, dental, and vision benefits package
- 401(k) match (U.S.-based employees only)
- $200/month Health & Wellness stipend
- Continuing Education support
- $500/year Function Health subscription (U.S.-based employees only)
- Free parking for in-office employees
- Flexible Time Off (FTO)
- Parental leave for eligible employees
- Supplemental life insurance
webAI is an Equal Opportunity Employer and does not discriminate against any employee or applicant on the basis of age, ancestry, color, family or medical care leave, gender identity or expression, genetic information, marital status, medical condition, national origin, physical or mental disability, protected veteran status, race, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable laws, regulations and ordinances. We adhere to these principles in all aspects of employment, including recruitment, hiring, training, compensation, promotion, benefits, social and recreational programs, and discipline. In addition, it is the policy of webAI to provide reasonable accommodation to qualified employees who have protected disabilities to the extent required by applicable laws, regulations and ordinances where a particular employee works.
Skills
- Machine Learning
- PyTorch
- TensorFlow
- Hugging Face Transformers
- Retrieval-Augmented Generation
- Vector Databases
- Computer Vision
- Chroma
- FAISS
- Milvus
- Pinecone
- CUDA
More jobs at webAI
All 16Senior Product Manager, Developer Platform
webAI · Austin, TX · 21d ago
Director, Application Security
webAI · Austin, TX · 23d ago
Peer to Peer Software Engineer
webAI · Austin, TX · 1mo ago
Senior Rust Platform Engineer
webAI · Austin, TX · 1mo ago
Embedded Signal Processing Engineer
webAI · Remote Office · 1mo ago
Similar roles
Lead Software Engineer - City
Okc · Oklahoma City, OK, United States · USD 40–60/hr · today
Full-stack Software Engineer III/Senior, AI Gateway
Ngrok · US · San Francisco, CA · USD 202,500–247,500/yr · today
Senior Software Engineer
Redwood Materials · remote · Nevada · USD 180,000–237,500/yr · today
Embedded Software Engineer – Power Electronics, Energy Storage
Redwood Materials · San Francisco, California, United States · USD 180,000–237,500/yr · today
Senior Software Engineer (Reston or Cambridge)
Akamai · United States · USD 121,400–218,600/yr · today
Software Engineer
Realtor.com Careers · Austin · today