AI Engineer
Jade Global
- San Jose, United States +1
- Mid Level
- Posted
Skills
Required
- Python
- PyTorch
- LoRA
- LLMs
- CUDA / GPU
- data ingestion/validation pipelines
- vLLM / LLM
- inference-serving
Preferred
- Azure
- Blob
- Storage
- Snowflake
- training pipeline
- data pipeline
- GPU infrastructure
- PEFT
- AI Engineer1 We're building a self-service LoRA / full fine-tuning platform for LLMs, deployed on AKS.
- Basic Kubernetes familiarity
Responsibilities
- This role owns the training pipeline and data pipeline — turning raw datasets into validated training inputs and running LoRA/full fine-tuning jobs on GPU infrastructure.
- What you'll do Build and improve LoRA and full fine-tuning workflows (PyTorch + PEFT + CUDA) Own dataset ingestion, transformation, and validation
- Work within existing config and logging contracts shared with other platform teams
Requirements
- What you need
- Strong Python and hands-on PyTorch experience
- Experience with LoRA or full fine-tuning of LLMs Comfort debugging CUDA/GPU training workloads
- Experience building data ingestion/validation pipelines
- Nice to have vLLM or LLM inference-serving experience
Preferred
- Azure (Blob Storage, Snowflake) experience
jobapplier