Remote ML Engineer — AI Pipelines & Fine-Tuning

Baasi; Austin (TX); Hybrid; USD 100,000 - 140,000; Full time

Benefits offered by this job

Competitive compensation

Equity potential

Flexible remote work

Job summary

A stealth-mode startup in Austin is seeking an AI/ML Engineer (Python) to design and implement AI pipelines. This role focuses on productizing and automating fine-tuning techniques, ensuring vendors can manage their own adapters effortlessly. You will implement fine-tuning pipelines, automate data preprocessing, and write tests for reproducibility. The ideal candidate has strong Python skills, hands-on experience with PyTorch, and familiarity with fine-tuning methods. The company offers competitive compensation and a flexible remote work model.

Qualifications

• Strong programming skills in Python.
• Familiarity with LoRA/QLoRA or parameter-efficient fine-tuning methods.
• Ability to collaborate with backend/frontend engineers who are not ML specialists.

Responsibilities

• Implement and maintain LoRA/QLoRA fine-tuning pipelines using PyTorch.
• Automate data preprocessing for user-supplied datasets.
• Write tests to guarantee reproducibility and correctness of adapter outputs.

Skills

Python

Hands-on experience with PyTorch

Familiarity with LoRA/QLoRA

Understanding of mixed precision training

Experience building production-ready training scripts

Tools

Hugging Face ecosystem

Linux GPU environments

Job description

A stealth-mode startup in Austin is seeking an AI/ML Engineer (Python) to design and implement AI pipelines. This role focuses on productizing and automating fine-tuning techniques, ensuring vendors can manage their own adapters effortlessly. You will implement fine-tuning pipelines, automate data preprocessing, and write tests for reproducibility. The ideal candidate has strong Python skills, hands-on experience with PyTorch, and familiarity with fine-tuning methods. The company offers competitive compensation and a flexible remote work model.