Member of Technical Staff, Agentic Environments

🔥 0 minutes ago

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Cohere

WebsiteLinkedInAll Job Openings

11 - 50 employees

🤖 Artificial Intelligence

🏢 Enterprise

☁️ SaaS

Artificial Intelligence • Enterprise • SaaS

Cohere is a leading enterprise AI platform optimized for generative AI, search and discovery, and advanced retrieval. The company offers AI-powered applications designed to augment and elevate the global workforce, helping businesses thrive in the AI era. Cohere provides solutions such as embedding and reranking models, allowing enterprises to efficiently retrieve information and build powerful applications. The company offers flexible deployment options for enterprise-grade AI, on any cloud or on-premises, and provides extensive developer resources and support. Cohere is committed to scaling intelligence to serve humanity, making intelligence abundant, affordable, and accessible.

📋 Description

• Design and write high-performing, scalable software for training models • Develop tools to support and accelerate research and LLM training • Coordinate with Infrastructure, Efficiency, Serving, and scientific teams including Agent, Multimodal, and Multilingual • Research, implement, and experiment with ideas on cluster and data infrastructure • Collaborate with scientists, engineers, and other teams • Work on production systems and experiment with frontier models • Design experiments and design documents for scalable, production-ready solutions

🎯 Requirements

• Extremely strong software engineering skills • Value test-driven development, clean code, and reducing technical debt • Proficiency in Python • Proficiency with ML frameworks such as JAX, PyTorch, and/or XLA/MLIR • Experience using and debugging large-scale distributed training strategies, including memory and speed profiling • Experience with distributed training infrastructure such as Kubernetes • Experience with associated frameworks such as Ray • Experience in ML and LLM academic research • Passion for quality work • Comfortable tuning and optimising large LLM models • Comfortable working with people with different levels of software engineering skills • Comfortable diving into complex ML codebases to identify and resolve issues • Ability to work in a fast-paced, technically challenging environment

🏖️ Benefits

• A weekly lunch stipend of $75/£75 or equivalent in your local currency • Full health and dental benefits, including a separate budget for mental health • RRSP matching, 401K, Pension Scheme • 100% Parental Leave top-up for up to 6 months, for either parent • Annual enrichment benefits for arts and culture, fitness/wellness, quality time, and workspace improvement • Education & learning stipend for conferences, courses, and coaching • 6 weeks of paid vacation (30 working days) • Budget for traveling to other offices if you are remote • Annual company offsite • Daily lunch program, snacks, and regular community and social events for office-based employees • Co-working benefit for employees not near an office • $500 home office stipend