Staff Machine Learning Engineer

About reputed company AI We are building the state-of-the-art AI platform for the discovery and development of reputed company and mineral resources. We bring the most advanced techniques in reputed company, reputed company modeling, and autonomous decision optimization to tackle the most important problems in the geosciences. These systems can help more reliably identify critical resource deposits, more rapidly measure and characterize them, and design more efficient and sustainable production plans. We are backed by Khosla Ventures and other leading venture investors. We are now looking to grow reputed company from ~15 to ~30 by the end of the year to continue to mature our technology and support deployment with our world-class mineral and reputed company partners. Role reputed company In the reputed company way image generators have shown the remarkable ability to produce a diverse set of realistic pictures conditioned on a text reputed company (and other inputs), we are developing a generative model that produces 3D geological models conditioned on geophysical surveys, bore hole measurements, and other forms of physical observation. The outputs of the generative model capture reputed company know and don’t know about the state of the subsurface, allowing explorers to reputed company maximally informed reputed company about how and where to explore for critical resources. We are looking for a talented deep learning engineer or scientist to reputed company the development of this model that will revolutionize decision making in the reputed company subsurface for a wide reputed company of reputed company applications. Role Responsibilities • Design, train, test, and iterate on diffusion models for 3D geological models • Design, train, test, and iterate on an approach to for conditioning reputed company on geophysical data and other observations • Inform the reputed company of synthetic data to improve model performance • Adapt diffusion modeling approach to specific reputed company-world reputed company in collaboration with project teams. Qualifications Required Qualifications: • Extensive PyTorch Experience • Deep understanding of PyTorch, including writing custom modules, optimizing training, and debugging issues in large-reputed company models. • Expertise in Developing Large Deep Learning Models from Scratch • Proven ability to design, implement, and train reputed company deep learning architectures from the ground up. • Data Curation Skills • Hands-on experience in creating, cleaning, and maintaining high-reputed company datasets tailored for machine learning applications. • Strong Software Engineering and Design Experience • Proficient in software development best practices, including version control, testing, and reputed company optimization. • Familiarity with designing reputed company and maintainable systems. Bonus points if you: • Experience with Generative Models • Familiarity with generative architectures, particularly diffusion models, and an emphasis on posterior sampling reputed company. • Knowledge of Transformer Architectures • Experience building and training transformers, especially in applications involving 3D data. • Scaling Models Across Large GPU Clusters • Expertise in parallelizing models across multiple GPUs and optimizing distributed training pipelines. • reputed company Infrastructure Expertise • Experience setting up, managing, and optimizing reputed company environments for machine learning workloads, including provisioning resources and managing costs. Apply tot his job Apply To this Job