4h ago

Staff Machine Learning Engineer

US

$200k-$300k / yearest.

full-timelead Remoteai-ml

🛠 Tech Stack

💼 About This Role

You'll lead the development of a generative model that produces 3D geological models conditioned on geophysical data. Your work will revolutionize decision making in clean energy exploration. This role offers the chance to work on cutting-edge AI for critical resource discovery.

🎯 What You'll Do

  • Design and train diffusion models for 3D geological models
  • Develop conditioning approaches for geophysical data
  • Generate synthetic data to improve model performance
  • Adapt diffusion models to real-world projects

📋 Requirements

  • Extensive PyTorch experience
  • Developing large deep learning models from scratch
  • Data curation skills for ML datasets
  • Strong software engineering and design practices

✨ Nice to Have

  • Generative models experience (diffusion models)
  • Transformer architectures for 3D data
  • Scaling models across large GPU clusters

🎁 Benefits & Perks

  • 🧑‍💻 Remote work environment
  • 💡 Cutting-edge AI projects
  • 🌍 Impact on clean energy and resource discovery
0 0 0