> Markdown version of [/jobs/ext/2534763-machine-learning-scientist-drug-discovery](https://www.wearedevelopers.com/jobs/ext/2534763-machine-learning-scientist-drug-discovery). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md). --- # Machine Learning Scientist - Drug Discovery - **Company:** AstrixTechnology LLC - **Location:** South San Francisco, CA, United States - **Salary:** $124,800.0 - $145,600.0 - **Contract:** Temporary to permanent - **Skills:** Artificial Intelligence, Computer Vision, Clinical Data Repository, Cloud Computing, Computational Biology, Data Structures, Software Debugging, General-Purpose Computing on Graphics Processing Units, Python (Programming Language), Machine Learning, Natural Language Processing, Tensorflow, Software Engineering, Cloud Platform System, Pytorch, Large Language Models, Deep Learning, Information Technology, ONNX (Open Neural Network Exchange) Format, Machine Learning Operations, Software Version Control, Data Pipelines - **Published:** August 16, 2026 - **Apply:** https://www.nexxt.com/job.asp?id=3355997556&tx=CT3836THI&pt=1&aff=0B19D771-A501-4A5E-8338-2A822B784D54&utm_source=Job%20Feed&utm_medium=textkernel&utm_campaign=DE&utm_term=0B19D771-A501-4A5E-8338-2A822B784D54 ## About the Role * B.S., M.S., or Ph.D. in Computer Science, Machine Learning, Computational Biology, Data Science, Statistics, Mathematics, or a related quantitative discipline. * Must be authorized to work in the United States without current or future employer sponsorship. * 1-5 years of relevant professional experience, including postdoctoral research where applicable. * Strong foundation in data structures, algorithms, software engineering, and computational problem solving. * Expert-level Python programming skills. * Extensive experience with deep learning frameworks such as PyTorch, JAX, or TensorFlow. * Strong debugging and software development skills, with the ability to independently diagnose and resolve complex technical issues. * Experience with large-scale or distributed model training, such as DDP, Ray, FSDP, or DeepSpeed. * Experience with model deployment technologies such as Triton or ONNX. * Experience working with cloud and/or GPU computing infrastructure. * Hands-on experience with geometric deep learning, molecular cofolding models, neural force fields, or related scientific ML approaches is highly preferred. ## Description This role will focus on designing, developing, training, and deploying advanced machine learning models and computational engines that support lab-in-the-loop molecular design and optimization . Areas of focus include sequence modeling, molecular structure, conformational ensembles, molecular property prediction, natural language processing, computer vision, and robotics. The successful candidate will work in a highly collaborative, multidisciplinary environment alongside ML scientists, ML engineers, computational scientists, and drug design experts to develop next-generation solutions at the intersection of AI and life sciences., * Design, develop, optimize, evaluate, and deploy advanced deep learning models, including large language models, multimodal transformers, and generative AI models. * Build and optimize scalable data pipelines supporting machine learning and scientific applications. * Optimize model training and inference for performance, scalability, and accuracy using multi-GPU and cloud-based infrastructure. * Develop and maintain MLOps workflows covering model deployment, version control, monitoring, reproducibility, and ongoing model performance. * Develop machine learning approaches that connect diverse datasets, including genomics, transcriptomics, imaging, molecular, and clinical data. * Partner with scientists and engineers across disciplines to translate innovative machine learning methods into practical applications for drug discovery, disease research, and biomedical applications. * Independently troubleshoot complex modeling, software, and infrastructure challenges and drive solutions from development through deployment. ## Related Videos - [Getting Started with Machine Learning](https://www.wearedevelopers.com/videos/260-getting-started-with-machine-learning) - [Geometric deep learning for drug discovery](https://www.wearedevelopers.com/videos/264-geometric-deep-learning-for-drug-discovery) - [Machine learning in the browser with TensorFlowjs](https://www.wearedevelopers.com/videos/155-machine-learning-in-the-browser-with-tensorflowjs) - [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding) - [Introduction to Azure Machine Learning](https://www.wearedevelopers.com/videos/368-introduction-to-azure-machine-learning) - [30 Golden Rules of Deep Learning Performance](https://www.wearedevelopers.com/videos/11-30-golden-rules-of-deep-learning-performance) ## Related Articles - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [What Are Large Language Models?](https://www.wearedevelopers.com/magazine/304-what-are-large-language-models) - [MLOps – What’s the deal behind it?](https://www.wearedevelopers.com/magazine/125-mlops-what-s-the-deal-behind-it) - [Coffee with Developers - Maria Apazoglou - Making AI understandable for all in production](https://www.wearedevelopers.com/magazine/475-coffee-with-developers-maria-apazoglou-making-ai-understandable-for-all-in-production) - [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development) - [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers)