Senior / Lead ML Tooling Software Engineer

**This role requires eligibility to work in the USA for a US company and infrequent travel may be required**

Position Overview:
As the Senior / Lead ML Tooling Software Engineer at Atlas AI, reporting to the Head of Engineering, you will be part of the core team defining, building, testing, and delivering our GeoAI platform to enable internal and external Data Scientists (DS) and Machine Learning Engineers (MLE) to implement Geospatial AI models. The Senior / Lead ML Tooling Software Engineer position sits in our Software Engineering team; you will interface with peer software engineers, domain experts, data scientists, machine learning engineers, product designers, and product marketers. You will work at the intersection of our leading AI research activities and software engineering to advance our product roadmap.
As the Lead ML Tooling Software Engineer, you will:
Own the design, development, deployment and support of our cloud-scale training services for GeoSpatial Machine Learning models
Be the Subject Matter Expert ensuring that the Software Engineering team builds and delivers solutions that “just work” for how DS and MLE approach GeoSpatial problems in the real world
Engage internal collaborators and our end-use partners and customers to understand needs, plan solutions and lead development of deliverables
Close the feedback loop with Solutions, Engineering and Innovation teams to deliver customer experience and market insights
Architect and develop serverless solutions that marry products within and across cloud platforms (e.g. GCP, AWS, Azure) that address customers’ needs
Lead client-oriented API design and development
Work with designers, product managers, and your fellow engineers to build APIs to support key functionality and use cases
Actively participate to improve engineering standards, tooling, and processes
Leverage your experience in early stage startups to make informed decisions that balance speed vs quality
Focus on designing architectures to solve core problems, not just because it is the latest trend
What you’ll bring:
Three or more years of recent work experience as a practicing Data Scientist or Machine Learning Engineer using PyTorch, sklearn and lightgbm
Two or more years of recent work experience as a practicing Backend Software Engineer using Python, GDAL and other GeoSpatial tools deployed in the cloud as serverless applications (e.g. GCP Cloud Run) or orchestrated workflows (e.g. Airflow, Argo)
At least three or more years of recent work experience building around Earth Observation use-cases
Technical ownership of key areas of a Product, such as one or two critical services
Demonstrable experience with Python, Pandas, PyTorch, GDAL
Experience with building distributed computing infrastructure flexible for both on-demand and batch processing
Self-starter attitude, comfortable working independently and in an entrepreneurial, start-up environment; thrives in high-trust environments with minimal resources/backing
Exceptional interpersonal communication, both written and oral
Independence and ownership of work in an entrepreneurial, start-up environment
Intellectual curiosity and and desire to contribute across cultures in a dynamic global environment
Experiences that are big pluses:
Applying Geospatial libraries and datasets at scale for user-facing applications
Developing enterprise PaaS and SaaS products
Working in early-stage startup environments
Developing on Google Cloud Platform
Comfort working in a distributed team environment.
Please provide a link to your github repo in your CV.
We are ideally looking for a candidate in the SF Bay Area, CA or New York, NY (100% remote with two to four meet-ups per year in-person). We offer a casual and inclusive working environment, competitive compensation, and great benefits. We are a values-driven company, and take great pride in building products that are designed to improve lives and foster development across emerging markets.
Salary range for candidates depending upon level of experience and location: $170,000 – $200,000.
Company Overview:
At Atlas AI, we are pioneering the use of advanced geospatial data and AI to tackle some of the most pressing challenges of our time. Our mission is to empower organizations to adapt and thrive amidst changing global forces such as climate change. We provide a cutting-edge data science platform that offers a living map of population, wealth, climate, and economic development trends worldwide. This enables organizations to harness real-time data to build geospatial AI models, guiding investment decisions and fostering resilience in an uncertain future.
Our technology addresses two critical gaps: the lack of granular, up-to-date information on local development trends, and the difficulty most organizations face in building predictive models with geospatial data. By seamlessly integrating these capabilities into enterprise workflows, Atlas AI helps organizations navigate climate uncertainty, support vulnerable communities, optimize supply chains, and target investments with precision. Our customers have leveraged our platform to accelerate solar market adoption, forecast demand for climate adaptation solutions, and enhance nutrition supply chains in climate-vulnerable regions.
Why Join Us:
At Atlas AI, you’ll be part of a dynamic team dedicated to making a tangible impact on the world. We bring together world-class scientific leadership in AI with top-notch engineering and data science disciplines to build rigorous, scalable systems for large enterprise customers. Our collaborative environment fosters innovation and empowers you to work on groundbreaking projects that drive real change, from advancing sustainable energy solutions to optimizing global supply chains for vulnerable communities.
We are a highly values-driven organization, guided by our core principles of Integrity, Innovation, Impact, Collaboration, and Empowerment. If you are passionate about using technology to solve critical global challenges and are eager to contribute to a mission-driven company, Atlas AI is the place for you. Join us and be at the forefront of innovation, helping to build a more resilient and equitable future.