I’m supporting one of Japan’s fast-growing AI companies, with a highly international AI engineering and research team based in Tokyo. They are hiring an LLM Research Engineer to work on frontier AI Agent problems, including reasoning, long-term memory, multi-agent systems, evaluation, and large-scale model optimization.
¥12M–20M + Stock Options Tokyo | Hybrid Highly international AI team Python / PyTorch / JAX / Transformers / vLLM Claude Code MAX / Cursor / ChatGPT / Devin provided
The company is building enterprise AI Agents that can connect across SaaS products and execute complex business workflows autonomously.
What you’ll work on
Agent reasoning, planning, search, and tool use
Long-term memory and context compression
Multi-agent architectures and agent communication
Multimodal and long-context systems
LLM training, inference, and data optimization
Agent evaluation and quantitative benchmarking
Inference latency and cost optimization
Reproducing and improving state-of-the-art research
Who they’re looking for
Strong background in LLMs, Machine Learning, and Software Engineering
Experience building complex LLM-based agent systems
Hands-on experience with PyTorch or JAX
Deep understanding of Transformers and modern LLM architectures
Ability to read research papers and turn them into working implementations
Strong Python engineering skills
Master’s or PhD in CS, AI, ML, Mathematics, Physics, or a related field
Experience with large-scale RL, RLHF / DPO, multi-agent systems, multimodal models, agent evaluation, or publications at NeurIPS / ICML / ACL / EMNLP is a strong plus. For researchers and engineers who want to work on advanced LLM technology in Tokyo without giving up an international working environment, this is a particularly interesting opportunity.
If you’re working on LLMs, AI Agents, RL, model training, or agent evaluation and are interested in joining an international AI team in Tokyo, feel free to DM me for the full JD. #LLM #ResearchEngineer #AIAgent #AIResearch #MachineLearning #DeepLearning #TokyoJobs #JapanTechJobs #Hiring