Technical Staff Member

This is a fully remote position, open to applicants in United States, +2 more locations.

📋 Description

• Take charge of systems that support drug discovery research, encompassing scientific data pipelines, model training and evaluation, inference, compute infrastructure, and developer tooling.

• Collaborate with researchers to pinpoint critical challenges and guide solutions from exploration to implementation and continuous operation.

• Incorporate datasets into training processes and merge models, datasets, and evaluation techniques while maintaining scientific integrity and data accuracy.

• Enhance GPU workload efficiency, shared compute performance, cost-effectiveness, and reliability through effective scheduling, monitoring, failure recovery, and artifact management.

• Develop APIs to facilitate scientific results, libraries, automation, and dependable scientific workflows.

• Detect and eliminate bottlenecks in data preparation, execution, and evaluation processes.

• Recognize engineering gaps that hinder research advancement, prioritize with the team, and illustrate the impact of solutions.

⛳️ Requirements

• Proficient in Python and foundational software engineering principles, including data structures, interface design, testing, concurrency, and systematic debugging.

• Proven experience in managing significant software systems from conception to deployment and ongoing maintenance, including diagnosing failures and enhancing performance and reliability.

• Familiarity with building or supporting machine learning or scientific computing workflows, with insight into how data, model execution, and evaluation interconnect.

• Hands-on experience with PyTorch, particularly in debugging model execution and grasping how data loading, device placement, gradients, and memory utilization influence training and inference.

• Background in operating software on Linux, utilizing containers, and implementing maintainable changes through automated testing and CI/CD practices.

• Capability to quickly learn new systems, make independent technical decisions, and communicate reasoning and trade-offs effectively.

• Skill in managing issues across application code, data pipelines, machine learning execution, and infrastructure while gaining the necessary depth to resolve them.

• Competence in utilizing AI development tools effectively and taking ownership of the correctness and maintainability of the resulting code.

• Experience with Kubernetes, cloud platforms, workflow orchestration, databases, or computational biology and chemistry is advantageous.

• A background in biology or chemistry is not mandatory.

🏝️ Benefits

• Competitive salary and benefits package.

• Opportunities for professional development and growth within the organization.

• Collaborative and innovative work environment.