Senior Machine Learning Engineer

Department: Engineering

Location: Austin, TX

Employment Type: FullTime

About Us:

We are establishing the first distributed Al infrastructure dedicated to personalized Al.
The evolving needs of a data-driven society are demanding scalability and flexibility. We believe that the future of Al is distributed and enables real-time data processing at the edge, closer to where data is generated. We are building a future where a company's data and IP remains private and it's possible to bring large models directly to consumer hardware without removing information from the model.

What’s the Job?

We are seeking a Senior Machine Learning Engineer with expertise in Large Language Models (LLMs) and Natural Language Processing (NLP) to join our innovative team. This role will focus on developing and deploying state-of-the-art NLP models that power webAI’s platform. If you have experience in working with transformer models, language generation, and advanced text analytics, this is the perfect opportunity to apply your skills on groundbreaking AI technology.

You’ll collaborate closely with our research and engineering teams to design, build, and optimize models that work with both structured and unstructured language data. The ideal candidate will bring experience deploying production-level NLP models, as well as expertise in MLOps practices and scalable cloud infrastructure.

Key Responsibilities:

• Design, build, and fine-tune LLMs and NLP models (e.g., GPT, BERT) for tasks like language understanding and generation.

• Extend webAI’s existing LLM frameworks and libraries, incorporating the latest research in language models.

• Preprocess and prepare text datasets for training, including tokenization and feature extraction.

• Implement MLOps best practices for deploying scalable, production-level models on cloud platforms.

• Collaborate with cross-functional teams to integrate LLM models into the webAI platform.

• Stay updated on LLM advancements and apply the latest research to improve model performance.

Required Skills & Experience:

• 5+ years of experience in machine learning, with a focus on NLP and LLMs.

• Bachelor’s (Master’s preferred) in Computer Science, Machine Learning, or related field.

• Expertise in transformer architectures (e.g., GPT, BERT), text preprocessing, and feature engineering.

• Proficiency in Python and NLP libraries (Hugging Face, TensorFlow, PyTorch).

• Experience with MLOps tools (MLFlow, Kubeflow, etc.).

• Strong communication skills and passion for learning.

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 ad learning-oriented mindset, acknowledging the strengths and contributions of others.

Benefits:

• 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, political affiliation, 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.