Member of Technical Staff, Agentic Environments

Cohere
EU, Canada, New York
Workplace: RemoteFull timeFunction: Software EngineeringSkills: ["Collaboration","Quality focus","Problem-solving","Clean code mindset","Fast-paced adaptability"]

Build and scale production software for training foundation models, bridging research concepts with real-world implementations. Develop tools that accelerate LLM training and experiments, and work closely with engineering teams (infrastructure, efficiency, serving) and scientific teams. You’ll research, implement, and experiment on cluster and data infrastructure, collaborating across scientists and engineers to ensure smooth large-scale distributed training operations.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Cohere
Cohere
11 hours ago

Member of Technical Staff, Agentic Environments

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 6 hours agoStatus: Live

Job Summary

Build and scale production software for training foundation models, bridging research concepts with real-world implementations. Develop tools that accelerate LLM training and experiments, and work closely with engineering teams (infrastructure, efficiency, serving) and scientific teams. You’ll research, implement, and experiment on cluster and data infrastructure, collaborating across scientists and engineers to ensure smooth large-scale distributed training operations.
Location: EU, Canada, New York
Workplace: Remote
Employment Type: Full time
Job Function: Software Engineering

Key Responsibilities

  • •Design and write scalable software for training models.
  • •Develop new tools to support and accelerate research and LLM training.
  • •Coordinate with infrastructure/efficiency/serving engineering teams and scientific teams (e.g., Agent, Multimodal, Multilingual).
  • •Research, implement, and experiment with ideas on cluster and data infrastructure.
  • •Collaborate with scientists and engineers across teams to deliver production-ready solutions.
Travel: Low travel

Pay and Benefits

Perks:Health InsuranceDentalRrsp Matching401kPension SchemeParental LeavePaid LeaveLearning BudgetHome OfficeTravel Allowance

Key Requirements

  • •Extremely strong software engineering skills and a focus on scalable, high-performing systems.
  • •Value test-driven development and clean code, working to reduce technical debt across the codebase.
  • •Proficiency in Python and ML frameworks including JAX, Pytorch, and/or XLA/MLIR.
  • •Experience using and debugging large-scale distributed training, including memory/speed profiling.
  • •Experience with distributed training infrastructure such as Kubernetes and frameworks like Ray.
Experience:Machine learningLLMDistributed trainingEnterprise AIAcademic research
Skills:CollaborationQuality focusProblem-solvingClean code mindsetFast-paced adaptability
Tech Stack:PythonJAXPytorchXLAMLIRKubernetesRayLLM

Company Brief

Cohere
Builds security-first foundation models and enterprise AI products (LLMs, retrieval, agent platforms) for regulated industries, enabling customizable, private deployments across cloud and on-premises for real-world business applications.
Industry: AI & Machine Learning
Company Size: Large (251 to 1,000 employees)
Growth: Scaleup
Valuation: Unicorn (USD 1B+)
Funding: Series D
Headquarters: Toronto, Canada
Founded: 2019
Glassdoor
Glassdoor: 2.9
WebsiteLinkedInGlassdoor