Member of Technical Staff, Agent Code

About the Role

The Agent Code team operates primarily between ET to CET time zones, and you will be collaborating with a team of talented engineers and researchers to build scalable, production-ready solutions. Your work will have a significant impact on the widespread adoption of AI, and you will be playing a crucial role in shaping the future of AI systems.

Cohere's mission is to scale intelligence to serve humanity, and as a member of the technical staff, you will be instrumental in increasing the capabilities of Cohere's models and the value they drive for customers. You will be working with world-class compute resources, data, and talent to ensure you can do your best work.

What You Will Do

• Stay up-to-date with the latest research in code LLMs, agents, and related fields, implementing novel ideas into Cohere's systems
• Design and implement scalable strategies to train code models, and deploy agent frameworks for inference and sampling
• Collaborate with the pretraining team to create SFT trajectories and work on existing and new RL algorithms
• Hillclimb on existing benchmarks and design new ones that reflect the needs of Cohere's enterprise users
• Lead experiments on state-of-the-art compute infrastructure, pushing the boundaries of what's possible with frontier LLMs
• Develop and maintain large-scale codebases, ensuring they are efficient, scalable, and well-documented
• Work with cross-functional teams to prioritize and deliver high-impact projects
• Participate in code reviews, ensuring high-quality code and contributing to the improvement of the overall codebase
• Collaborate with researchers to develop and apply new techniques to real-world problems

What We Are Looking For

• A PhD in Computer Science, Machine Learning, or a related field, with publications in top-tier venues (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP)
• Deep expertise in code LLMs and agent systems, with a strong understanding of the latest research and trends
• Hands-on experience with frontier LLMs and their applications in code generation or automation
• Strong software engineering skills, with proficiency in Python and PyTorch, TensorFlow, or similar frameworks
• Experience with distributed systems, cloud infrastructure, and scalable architectures
• A proactive, self-motivated mindset, with a passion for solving ambitious, open-ended problems
• Excellent communication and collaboration skills, with the ability to work effectively in a remote environment
• A strong understanding of the latest developments in AI, machine learning, and related fields

Nice to Have

• Experience with natural language processing, computer vision, or other related areas
• Familiarity with Agile development methodologies and version control systems (e.g., Git)
• Experience with cloud-based services (e.g., AWS, GCP, Azure)
• Knowledge of containerization (e.g., Docker) and container orchestration (e.g., Kubernetes)
• Experience with testing and validation frameworks

Benefits and Perks

• The opportunity to work on cutting-edge problems at the intersection of AI, code generation, and autonomous agents
• Access to world-class compute resources, data, and a collaborative team of researchers and engineers
• A remote-friendly, flexible work environment with a focus on impact and innovation
• Competitive compensation and benefits, including equity in Cohere
• Opportunities for professional growth and development, including training and education programs
• A dynamic, supportive work environment with a strong focus on teamwork and collaboration
• Flexible working hours and a generous PTO policy
• Access to the latest tools and technologies, including cutting-edge hardware and software
• A chance to work with a talented team of professionals who are passionate about their craft

• Develop a strong portfolio showcasing your work in code LLMs and agent systems, including any relevant research or projects you have worked on.
• Familiarize yourself with Cohere's technology and research areas, and be prepared to discuss your understanding of the company's mission and goals.
• Practice explaining complex technical concepts in simple terms, as this will be an important part of your role.
• Be prepared to provide specific examples of your experience with frontier LLMs and their applications in code generation or automation.
• Highlight your ability to work independently and as part of a team, as well as your passion for solving ambitious, open-ended problems.
• Research the current market trends and salary ranges for similar roles, and be prepared to negotiate your compensation package.
• Be aware of the potential risks and challenges associated with working in a remote environment, and be prepared to discuss your strategies for staying motivated and connected with the team.