Machine Learning Research Engineer | Kilby Labs

Change the world. Love your job.

We are seeking a highly motivated Machine Learning Research Engineer to join our Embedded AI team to work on cutting‑edge large language model (LLM) research and development for edge AI applications.

• 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 systems for complex task automations.
• Collaborate with system teams and internal business teams to define and implement AI/ML solutions for core business.

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

• Strong background in Natural Language Processing, Large Language Models, and Deep Learning frameworks.
• Proven track record of designing, developing, and deploying machine learning models or 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.

• Engineer your future. We empower our employees to truly own their career and development.
• We value diverse backgrounds and perspectives that drive innovation.
• Competitive pay and benefits designed to help you and your family live your best life.

Texas Instruments is an equal opportunity employer and supports a diverse, inclusive work environment. All qualified applicants will receive consideration for employment without regard to race, color, religion, creed, disability, genetic information, national origin, gender, gender identity and expression, age, sexual orientation, marital status, veteran status, or any other characteristic protected by federal, state, or local laws.