ML Research Scientist, Foundation Models (Senior / Staff / Principal)
Benefits:
Competitive compensation package that includes salary and equity
Comprehensive health benefits: Medical, Dental, and Vision (covered 100% for the employees)
401(k) plan
Open (unlimited) PTO policy
Free lunches and dinners at our offices
Education Requirements:
PhD in machine learning, computer science, other computational sciences or equivalent research experience (nice to have)
Experience Requirements:
A portfolio of novel research in one or more cutting-edge domains of generative or predictive modeling (e.g., diffusion, flow matching, RL, foundation model architectures and training methods)
Hands-on experience with Pytorch, Pytorch Lightning, Ray Distributed Training, Pytorch Geometric, etc. (nice to have)
Experience in distributed training and inference of large models on GPU clusters (nice to have)
Other Requirements:
A strong coder, comfortable with going deep into the engineering stack to build, debug, and ship your own models
An independent, first-principles thinker with a strong sense of ownership
A true team player with strong communication skills
Familiarity with molecular data, (proteins, small molecules), physics-informed ML, or 3D point cloud data (nice to have)
Responsibilities:
Lead transformative research projects from conception to implementation, tackling core challenges in generative modeling for molecular systems
Design and develop novel models and algorithms, building upon the latest literature in diffusion models, flow matching, RL, LLMs, and other cutting-edge areas
Execute ambitious experiments at scale, leveraging our world-class computational infrastructure to rigorously validate your hypotheses
Collaborate intensely with a multidisciplinary team to translate your research into tangible impact on our drug discovery platform
Contribute to the global research community by publishing some of your work and representing Genesis at top tier AI/ML conferences
Show more details