Member of Technical Staff [Research]
NeoCognition · Palo Alto, CA · 2 mo ago
On-siteEngineeringFull-time
Responsibilities
- Lead research initiatives in the areas of LLM reasoning, post-training, and agentic system design.
- Develop new methods to improve capability, reliability, and safety of autonomous LLM agents in real-world environments.
- Collaborate with software and platform engineers to prototype and productionize research outcomes into sticky product experiences.
- Design and execute experiments, benchmark performance, and analyze model behaviors to identify failure modes and opportunities.
- Stay abreast of emerging work in reasoning, multi-agent systems, RLHF, tool use, and LLM fine-tuning — and contribute to publications or open-source efforts where appropriate.
- Help shape the research culture and technical roadmap of the company as an early member of the team.
Qualifications
- Strong background in machine learning, natural language processing, or AI systems, with experience in large language models (LLMs).
- Deep understanding of one or more of the following areas: Agentic system design (tool-use, planning, reasoning, computer-use), LLM post-training (instruction tuning, RL, reasoning), Data pipeline design and model evaluation.
- Proficiency in Python and familiarity with modern ML frameworks (e.g., PyTorch, JAX, or TensorFlow).
- Demonstrated ability to design, execute, and analyze research experiments — from idea to implementation.
- Strong communication skills and ability to work collaboratively in a fast-paced, cross-disciplinary environment.
- Nice to have:
- Experience working with open-weight models (e.g., Llama, Mistral, or similar) and training infrastructure.
- Publications in top-tier AI venues (NeurIPS, ICLR, ICML, ACL, etc.).
- Prior experience building research prototypes into usable tools or products.