Machine Learning Research Engineer | Kilby Labs

118-WW TMG MFG OPS; United States; On-site; USD 140,000 - 200,000; Full time

Job summary

118-WW TMG MFG OPS is seeking a MachineLearningResearchEngineer to lead LLM research for edge AI applications. You will explore architectures, agentic LLMs, and reasoning models to enable human‑like decision making for business impact.

The role involves collaborating with internal teams to implement AI/ML solutions. Minimum requirements include a Doctoral degree and 3 years of related experience, with strong skills in NLP, DL frameworks, and code in Python/C++.

Qualifications

• Doctoral degree in CS/EE or related field.
• 3 years of related experience.
• Strong background in NLP, LLMs, and DL frameworks.

Responsibilities

• Advance state-of-the-art LLM architectures and Agentic LLM designs.
• Research and develop efficient training and post-training algorithms.
• Design domain-specific knowledge integration and reasoning models.
• Collaborate with system and business teams to deploy AI/ML solutions.

Skills

Natural Language Processing

Large Language Models

Python

C/C++

Transformer models

Education

Doctoral degree in Computer Science or related field

Tools

PyTorch

JAX

ONNX

transformers

trl

vllm

Job description

Change the world. Love your job.

We are seeking a highly motivated MachineLearningResearchEngineerto join our Embedded AI team to work on cutting‑edge Large Language Model (LLM) research and development for Edge AI applications. As a key member of our team, you will lead the efforts on advancing the state‑of‑the‑art in LLM architectures, Agentic LLM, and Reasoning LLM. Your work will involve exploring innovative approaches to integrate LLMs with domain knowledge, related tools, and other AI techniques to achieve human‑like decision‑making capabilities for business impact.

In this machine learning research engineer role, you’ll have the chance to work on the following topics:

• Explore the application of LLMs in code generation, evaluation, and optimization
• Conduct research and development on novel and efficient LLM architectures, training algorithms, and post‑training algorithms
• Design and develop new reasoning models with domain‑specific knowledge
• Design and implement advanced Agentic LLM system for complex task automations
• Collaborate with system teams and internal business teams to define and implement AI/ML solutions for core business

Minimum requirements:

• Doctoral degree in Computer Science, Electrical Engineering, Electrical and Computer Engineering, or related field
• 3 years of related experience

Preferred qualifications:

• Strong background in Natural Language Processing, Large Language Models, and Deep Learning frameworks
• Proven track record of designing, developing, and deploying machine learning models/LLMs that drive business value
• Proficiency in Python, C/C++, and software design, including debugging, performance analysis, and optimization
• Excellent understanding of LLM architectures and transformer‑based models
• Experience with popular deep learning frameworks (e.g., PyTorch, JAX, ONNX) and LLM‑specific libraries (e.g., transformers, trl, vllm)
• Strong foundation in text processing, tokenization, and embedding techniques
• Knowledge of few‑shot learning, transfer learning, and fine‑tuning
• Knowledge of LLM performance evaluation
• Knowledge of reinforcement learning for LLM; Experience with LLM post‑training implementation, including PPO, DPO, and GRPO
• Experience with LLM agent implementation with tool calling
• Excellent communication and interpersonal skills, with the ability to work in a dynamic and distributed team