Senior / Lead ML Tooling Software Engineer

Remote
Full Time
Experienced

**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.

 

Perks & Benefits

  • Fast-paced and team-oriented environment where you will be instrumental in the direction of the company.
  • Atlas AI is a 100% remote company with occasional in-person meet-ups in SF, NYC, Nairobi (Kenya) and Kaunas (Lithuania).
  • Competitive compensation & equity.
  • Outsized responsibilities & professional development budget.
  • Opportunities to travel to conferences, events and other industry-led symposiums.
  • Training is foundational; functional, customer immersion, and development training.
  • Medical, dental, and vision insurance including dependents (exact benefits vary by region).
  • Unlimited paid time off, with a target goal of 20 days off per year + public holidays (region dependent).
  • Paid parental leave (exact benefits vary by region).
  • Home office setup allowance, co-working space stipend, and company MacBook.
  • Monthly wellness benefit.
  • Opportunities to co-author research papers and patents (role-dependent).
Share

Apply for this position

Required*
Apply with Indeed
We've received your resume. Click here to update it.
Attach resume or Paste resume
Attach resume as .pdf, .doc, .docx, .odt, .txt, or .rtf (limit 5MB) or Paste resume

Paste your resume here or Attach resume file

To comply with government Equal Employment Opportunity and/or Affirmative Action reporting regulations, we are requesting (but NOT requiring) that you enter this personal data. This information will not be used in connection with any employment decisions, and will be used solely as permitted by state and federal law. Your voluntary cooperation would be appreciated. Learn more.
Invitation for Job Applicants to Self-Identify as a U.S. Veteran
  • A “disabled veteran” is one of the following:
    • a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or
    • a person who was discharged or released from active duty because of a service-connected disability.
  • A “recently separated veteran” means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.
  • An “active duty wartime or campaign badge veteran” means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.
  • An “Armed forces service medal veteran” means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.
Veteran status


Voluntary Self-Identification of Disability
Voluntary Self-Identification of Disability Form CC-305
OMB Control Number 1250-0005
Expires 04/30/2026
Why are you being asked to complete this form?

We are a federal contractor or subcontractor. The law requires us to provide equal employment opportunity to qualified people with disabilities. We have a goal of having at least 7% of our workers as people with disabilities. The law says we must measure our progress towards this goal. To do this, we must ask applicants and employees if they have a disability or have ever had one. People can become disabled, so we need to ask this question at least every five years.

Completing this form is voluntary, and we hope that you will choose to do so. Your answer is confidential. No one who makes hiring decisions will see it. Your decision to complete the form and your answer will not harm you in any way. If you want to learn more about the law or this form, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

A disability is a condition that substantially limits one or more of your “major life activities.” If you have or have ever had such a condition, you are a person with a disability. Disabilities include, but are not limited to:

  • Alcohol or other substance use disorder (not currently using drugs illegally)
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS
  • Blind or low vision
  • Cancer (past or present)
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or serious difficulty hearing
  • Diabetes
  • Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders
  • Epilepsy or other seizure disorder
  • Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome
  • Intellectual or developmental disability
  • Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD
  • Missing limbs or partially missing limbs
  • Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports
  • Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)
  • Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities
  • Partial or complete paralysis (any cause)
  • Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema
  • Short stature (dwarfism)
  • Traumatic brain injury
Please check one of the boxes below:

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.

You must enter your name and date
Human Check*