AI Engineer

Jade Global • San Jose, CA

5d

About The Position

The AI Engineer will build and improve a self-service LoRA and full fine-tuning platform for LLMs deployed on AKS. The role owns the training and data pipelines, including converting raw datasets into validated training inputs and running LoRA and full fine-tuning jobs on GPU infrastructure. The engineer will work within existing configuration and logging contracts shared with other platform teams.

Requirements

• Strong Python skills and hands-on PyTorch experience.
• Experience with LoRA or full fine-tuning of LLMs.
• Comfort debugging CUDA/GPU training workloads.
• Experience building data ingestion and validation pipelines.

Nice To Haves

• vLLM or LLM inference-serving experience.
• Azure experience, including Blob Storage and Snowflake.
• Basic Kubernetes familiarity.

Responsibilities

• Build and improve LoRA and full fine-tuning workflows using PyTorch, PEFT, and CUDA.
• Own dataset ingestion, transformation, and validation.
• Run LoRA and full fine-tuning jobs on GPU infrastructure.
• Work within existing configuration and logging contracts shared with other platform teams.

What This Job Offers

• Job Type: Full-time
• Education Level: No Education Listed