ML Engineer - Austin, TX

Baasi is a stealth-mode startup building next-generation infrastructure for the AI industry. They are seeking an ML Engineer to help design and implement AI pipelines, focusing on productizing and automating fine-tuning techniques for vendors.

Responsibilities:

• Implement and maintain LoRA/QLoRA fine-tuning pipelines using PyTorch + Hugging Face Transformers + PEFT
• Develop logic for incremental training and adapter stacking, producing clean, versioned “delta packs.”
• Automate data preprocessing (tokenization, formatting, filtering) for user-supplied datasets
• Build training scripts/workflows that integrate with orchestration backends (Node.js, REST/gRPC, or job queues)
• Implement monitoring hooks (loss curves, checkpoints, eval metrics) to feed into dashboards
• Collaborate with DevOps to ensure reproducible, portable training environments
• Write tests to guarantee reproducibility and correctness of adapter outputs
• Willingness to occasionally be present in the office for discussions and team collaboration