LLM Research Engineer — Onsite Mountain View

Wal-Mart; Mountain View (CA); On-site; USD 158,000 - 180,000; Full time

Job summary

FocusKPI is seeking an LLM Research Engineer for a high-tech SaaS client in Mountain View, CA. This onsite, 12-month contract role centers on designing, training, and fine-tuning large language models and deploying them in production environments with MLOps practices.

You will work on cutting-edge NLP techniques, transformer architectures, and evaluation pipelines, contributing to open-source projects and research papers while collaborating with cross-functional teams.

Qualifications

• Master's degree with 3+ years of experience post-graduation or Bachelor's degree with 5+ years of experience post-graduation.
• Advanced degree in CS, AI, DS, or related field.
• Strong programming skills and ML proficiency.
• Proficiency with deep learning frameworks (TensorFlow, PyTorch, JAX).
• Experience with transformer-based models (GPT, BERT, RoBERTa, LLaMA).
• Familiarity with NLP and sequence-to-sequence models.
• Experience with Hugging Face libraries and OpenAI APIs.
• Experience with MLOps tools like Docker, Kubernetes, and CI/CD pipelines.
• Strong understanding of distributed computing and GPU acceleration (CUDA).
• Experience with evaluation pipeline development and LLM fine-tuning.
• Knowledge of RL and RLHF.

Responsibilities

• Design, train, and fine-tune large language models for various applications.
• Research NLP and ML techniques to improve model performance.
• Explore transformer architectures and multi-modal models.
• Prepare and preprocess large-scale text datasets.
• Develop data augmentation to improve training data quality.
• Ensure data bias is minimized and AI ethics are followed.
• Optimize model accuracy, efficiency, and scalability.
• Reduce latency and memory usage for real-time applications.
• Collaborate with MLOps to deploy LLMs in production with Docker/Kubernetes.
• Develop evaluation pipelines using accuracy, perplexity, BLEU, F1 scores.
• Test for bias, fairness, and robustness across datasets.
• Conduct A/B testing for model improvements in real-world tasks.
• Stay updated on generative AI, transformers, and NLP research.
• Contribute to papers, patents, and open-source projects.
• Present findings at conferences and internal sessions.

Skills

Strong programming

Machine Learning

Deep learning frameworks

Transformer models

NLP

Hugging Face

OpenAI APIs

MLOps tools

Docker

Kubernetes

CUDA GPU accel

Evaluation pipelines

RLHF

Education

Master's degree

Bachelor's degree

Tools

Docker

Kubernetes

CI/CD

Job description

FocusKPI is seeking an LLM Research Engineer for a high-tech SaaS client in Mountain View, CA. This onsite, 12-month contract role centers on designing, training, and fine-tuning large language models and deploying them in production environments with MLOps practices.

You will work on cutting-edge NLP techniques, transformer architectures, and evaluation pipelines, contributing to open-source projects and research papers while collaborating with cross-functional teams.