🛠 Tech Stack

💼 About This Role

You'll lead the development of generative AI models for 3D geological modeling to revolutionize clean energy and mineral exploration. You'll design, train, and iterate on diffusion models conditioned on geophysical data. This role offers the chance to build foundational models at a venture-backed startup tackling critical resource discovery.

🎯 What You'll Do

• Design and train diffusion models for 3D geological models.
• Develop conditioning approaches for geophysical data inputs.
• Inform synthetic data generation to improve model performance.
• Adapt diffusion models to real-world projects with project teams.

📋 Requirements

• Extensive PyTorch experience with custom modules and optimization.
• Proven ability to develop large deep learning models from scratch.
• Hands-on data curation skills for ML datasets.
• Strong software engineering and design experience.

✨ Nice to Have

• Experience with generative models, especially diffusion models.
• Knowledge of Transformer architectures for 3D data.
• Expertise in scaling models across large GPU clusters.

🎁 Benefits & Perks

• 🚀 Work on cutting-edge AI for clean energy and mineral discovery.
• 🏖️ Remote-first culture with flexible working hours.
• 💰 Competitive equity in a well-funded startup.
• 📈 Growth opportunities as team expands rapidly.
• 🛠️ Access to compute resources for large-scale model training.

📨 Hiring Process

Estimated timeline: 2-4 weeks · AI estimate

1. 1Recruiter Screen· 30 min
2. 2Technical Interview· 60 min
3. 3On-site (virtual) with team· 2 hours