remote
Machine Learning Research Engineer MLRE - GPUs - achira
ML Engineer
Lead end‑to‑end research and deployment of GPU‑accelerated ML models for AI‑driven chemistry, driving breakthroughs in molecular prediction and design on large‑scale distributed infrastructure.
About the role
Key Responsibilities
- Design, prototype, and optimize GPU‑centric machine learning models for molecular data.
- Collaborate with chemists and data scientists to translate chemical problems into scalable ML solutions.
- Implement and maintain distributed training pipelines on high‑performance compute clusters.
- Deploy production‑ready models, monitor performance, and iterate based on real‑world feedback.
- Document research findings and contribute to open‑source tooling and internal knowledge bases.
Requirements
- Strong programming skills in Python and experience with deep learning frameworks (PyTorch, TensorFlow).
- Proven expertise in GPU programming (CUDA, cuDNN) and distributed training techniques.
- Background in machine learning research, preferably with publications or significant projects.
- Familiarity with cheminformatics or computational chemistry concepts is a plus.
- Excellent problem‑solving abilities and a passion for pushing the boundaries of AI in science.
Skills
pythonmachine learningdeep learning