Senior Machine Learning Engineer – GeoAI Platform

About Wherobots

Wherobots was founded by the original creators of Apache Sedona to build the first fully‑managed, highly scalable geospatial cloud database and analytics platform:
Wherobots Cloud. Geospatial, location‑enabled, and satellite imagery data are quickly becoming a critical and valuable source of information and insights to a broad array of industries, from logistics and insurance to financial or climate tech companies. Wherobots helps those companies bring their geospatial data down to earth and drive value from it for their business and their customers through full‑featured and scalable computation, querying, analytics, and visualization capabilities.

Wherobots is looking for a passionate, skilled, and experienced Machine Learning Engineer to help architect, build, and operate the large‑scale geospatial ML platform that powers GeoAI workflows on hundreds of terabytes to petabytes of raster data.

This is a distributed‑systems‑first role with meaningful ML infrastructure ownership. You will spend most of your time building high‑throughput, GPU‑aware data pipelines that turn massive raster archives into features, predictions, and published outputs at global scale. The role sits at the intersection of distributed systems, ML inference, and geospatial data infrastructure. If you can design clean dataflow, get the most out of a GPU cluster, and turn research prototypes into resilient production systems, we should talk.

We are 100% cloud‑native and build our product using modern, reliable tooling. We use Ray, PyTorch, and the scientific Python stack (PyArrow, Num Py, Xarray) to operate on Zarr, Cloud‑Optimized GeoTIFF (COG), Geo Parquet, and Parquet data on object storage.

If you are passionate about building cutting‑edge ML infrastructure for the physical world and want to be part of a fast‑growing company at the forefront of geospatial technology, we would love to hear from you. Apply now and join the Wherobots team!

• Design and operate end‑to‑end ML pipelines
  :
  Build pipelines over massive raster archives such as Zarr and COG, from ingestion to feature generation to inference to publication.
• Build high‑throughput distributed pipelines
  :
  Use Ray (Datasets and actors) with careful control over I/O, compute overlap, and back pressure to keep clusters fully utilized.
• Optimize GPU inference at scale
  :
  Tune PyTorch inference pipelines using batching, CUDA stream overlap, and memory‑aware scheduling to maximize throughput per GPU.
• Develop spatial data processing patterns
  :
  Implement tiling, overlapping windows, and accumulators that match the access patterns of spatial models.
• Ensure production reliability
  :
  Build in retries, checkpointing, observability, and cost‑efficient scaling so long‑running global jobs are debuggable and resilient to failure.
• Build reusable platform abstractions
  :
  Collaborate on abstractions that generalize across datasets, models, and product use cases so new workflows ship quickly.
• Raise the bar
  :
  Provide technical leadership on architecture, engineering standards, and roadmap, and contribute to architecture and code reviews across the organization.

• 5+ years of experience building distributed data or ML systems in production.
• Strong hands‑on experience with Ray, Spark, Dask, or similar distributed computing systems.
• Deep understanding of performance tradeoffs across memory, I/O, serialization, and scheduling.
• Production experience running GPU inference workloads at scale.
• Proficiency in Python and the scientific Python stack (PyTorch, PyArrow, Num Py, Xarray).
• Strong working knowledge of object storage (e.g., S3) and large‑scale data access patterns such as range requests and sharding.
• Experience with columnar storage formats (Geo Parquet, Parquet) and chunked storage formats (Zarr v3).
• Geospatial experience, especially with raster data, projections, or tiling schemes.

• Experience designing data platforms or reusable ML infrastructure.
• Familiarity with the lakehouse architecture and table formats.
• Experience turning research or prototype ML workflows into production systems.

Wherobots offers competitive…