Machine Learning Engineer
SweedFull Time
Mid-level (3 to 4 years)
Candidates should possess a Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field. Strong proficiency in programming languages like Python is required, alongside a deep understanding of operating system internals, particularly Unix. Experience with high-performance computing (HPC) infrastructure, including Slurm and Kubernetes, is essential, as is familiarity with performance engineering tools and techniques such as profiling, optimizing CPU and GPU workloads, and writing CUDA and Triton kernels. Experience with large-scale ETL pipelines and deep learning libraries like PyTorch is also necessary.
The AI Engineer will collaborate with AI researchers to design, implement, and maintain software systems supporting AI-driven drug discovery applications. They will develop and optimize high-performance computing (HPC) infrastructure, integrate AI models into production systems and applications, ensuring scalability and reliability, and contribute to the overall AI-driven drug discovery pipeline.
Develops AI models for molecular predictions
Chai Discovery develops AI foundation models that focus on predicting and reprogramming interactions between biochemical molecules, which are crucial for life. Their main product, Chai 1, is a multimodal model that can predict various molecular structures, such as proteins, small molecules, DNA, RNA, and covalent modifications. This model is available for free through a web interface, making it accessible for both academic researchers and pharmaceutical companies, especially in the field of drug discovery. Unlike many competitors, Chai Discovery emphasizes a user-friendly platform that allows clients to easily integrate AI into their research and development processes. The company's goal is to enhance the efficiency of molecular research and drug discovery by providing powerful AI tools that can accelerate scientific advancements.