Machine Learning Engineer

Job Description:

1. Framework Migration: Contribute to the migration of fastai to Pytorch Lightning, working closely with the technical team to ensure compatibility with new and existing functionality.
2. Pipeline Enhancement: Design and implement a flexible pipeline architecture that maintains compatibility with existing features, with a focus on updated model architecture, advanced loss functions and ensemble predictions.
3. Polygon Enhancement: Design and implement improved polygon regularization techniques for accurate delineation of buildings, ensuring compatibility with evolving architecture.
4. Informal Settlement Detection: Integrate existing open-source solutions for informal settlement detection into the pipeline for building detection.
5. Testing Infrastructure: Update testing frameworks for both model performance and code quality, coordinating with the technical team to ensure coverage of new features.
6. MLOps Infrastructure: Implement DVC for dataset and model versioning alongside MLflow for experiment tracking and model registry.
7. Technical Documentation: Create detailed technical documentation covering architecture, deployment, and maintenance procedures.
8. Code Quality: Implement best practices for code organization, version control, and continuous integration across both new and existing components.
9. Integration Support: Work closely with the technical team to ensure all new infrastructure supports enhanced polygon regularization and multi-class detection needs.
10. Platform Compatibility: Ensure all infrastructure changes maintain compatibility with existing and new development tools.

Desired Experience:

1. Software Architecture: Experience designing and implementing large-scale ML systems.
2. Testing & CI/CD: Experience setting up testing frameworks and continuous integration pipelines.
3. Code Quality: Strong understanding of software engineering best practices and design patterns.
4. Team Collaboration: Demonstrated ability to work effectively in collaborative development environments and coordinate technical implementations across team members.
5. Documentation: Experience creating technical documentation and maintaining API documentation.

Languages:

1. English (required)

Qualifications:

1. Qualifications: Master's degree or higher in Computer Science, Software Engineering, or related field (or a Bachelor’s degree with equivalent work experience).
2. ML Engineering: At least 3 years of experience in data science or machine learning engineering, with focus on production systems.
3. Deep Learning Frameworks: Strong expertise in PyTorch or a related DL framework.
4. Geospatial Development: Familiarity with geospatial libraries and raster and vector processing (such as GDAL, rasterio, geopandas).