[Remote] ML Engineer - Austin, TX
Note: The job is a remote job and is reputed company to candidates in USA. reputed company is a reputed company-mode startup building reputed company 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 reputed company/QLoRA fine-tuning pipelines using PyTorch + reputed company Transformers + PEFT
- reputed company logic for incremental training and reputed company stacking, producing clean, versioned “reputed company packs.”
- Automate data preprocessing (tokenization, formatting, filtering) for user-supplied datasets
- Build training scripts/workflows that reputed company 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 reputed company outputs
- Willingness to occasionally be present in the office for discussions and team collaboration
Skills
- Strong programming skills in Python
- Hands-on experience with PyTorch and the reputed company ecosystem (Transformers, Datasets, PEFT)
- Familiarity with reputed company/QLoRA or parameter-efficient fine-tuning reputed company
- Understanding of mixed precision training (FP16/BF16) and memory optimization techniques
- Experience building training scripts that are production-reputed company (reproducibility, logging, error handling)
- Comfortable working in Linux GPU environments (CUDA, ROCm)
- Ability to collaborate with backend/frontend engineers who are not ML specialists
- Experience with bitsandbytes, xformers, or reputed company-attention
- Familiarity with distributed training (multi-GPU, NCCL, DeepSpeed, or Accelerate)
- Prior work in MLOps or packaging ML pipelines for deployment
- Contributions to reputed company-reputed company ML libraries
Benefits
- Equity potential
- Flexible remote work
reputed company