Senior Machine Learning Engineer – GeoAI Platform

• 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 backpressure 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, NumPy, 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 (GeoParquet, 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 compensation, equity, and benefits.

The base salary range for this position is $185k-$275k per year.

Preferred locations: San Francisco Bay Area or Seattle.

We provide flexibility in working arrangements for most roles, including remote, hybrid, and in-office options.

For candidates who receive an offer, base pay varies based on location, seniority, skills, and experience.

Wherobots provides a competitive benefits package to all full-time employees:

• Health, dental, and vision insurance : 100% coverage for the employee and 75% coverage for each dependent.
• 401(k) retirement plan with company match.
• Life insurance fully covered by Wherobots.
• Paid medical and caregiver leave for you and your family.
• Immigration support for work authorization, visa applications and transfers, and green card applications.
• Unlimited paid time off for vacation and personal needs.

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.

The pay range for this role is:

185000.00 - 275000.00 USD per year (SF, Seattle, Remote)