Machine Learning Research Engineer | Kilby Labs
Core
Lead research and development of Large Language Models (LLMs), Agentic LLMs, and reasoning models for Edge AI applications to achieve human-like decision-making.
Role type
Machine Learning Research Engineer (LLM/Edge AI)
Builds
Novel LLM architectures, training algorithms, post-training algorithms, and Agentic LLM systems for complex task automation.
Domain
Artificial Intelligence / Large Language Models / Edge Computing
Deliverable
production ML models
Required skills
LLM architecture design, training algorithms, post-training algorithms (PPO, DPO, GRPO), Agentic LLM system design, code generation and optimization, domain knowledge integration, transformer-based models, text processing, tokenization, embedding techniques, few-shot learning, transfer learning, fine-tuning, reinforcement learning, tool calling implementation
Preferred skills
Deep learning frameworks (PyTorch, JAX, ONNX), LLM libraries (transformers, trl, vllm), Python, C/C++, software design, debugging, performance analysis, optimization
Technologies
PyTorch, JAX, ONNX, transformers, trl, vllm
Responsibilities
Explore LLM applications in code generation, evaluation, and optimization; Conduct R&D on novel LLM architectures and algorithms; Design and develop reasoning models with domain-specific knowledge; Design and implement advanced Agentic LLM systems; Collaborate with system and business teams to define and implement AI/ML solutions.
Seniority
Mid-level, hands-on IC
