Biocube - Senior Machine Learning Engineer

Key Responsibilities :

- Design and manage the end-to-end ML pipeline, automating stages from data ingestion and model training to deployment and monitoring.

- Own and maintain CI/CD pipelines for deep learning models across multiple deployment targets : Android (TensorFlow Lite), iOS (CoreML), Web (Web Assembly), On- premises / edge devices (ONNX, TensorRT)

- Integrate and manage MLflow for tracking experiments, versioning models, and managing deployments.

- Leverage Docker and Kubernetes for scalable and reproducible ML workloads in development and production.

- Orchestrate workflows using Apache Airflow to ensure reliable, scheduled, and maintainable model lifecycle operations.

- Set up and maintain monitoring and alerting systems for model drift, inference performance, and system uptime.

- Automate model validation, conversion (TFLite/CoreML/ONNX), and performance benchmarking as part of CI pipelines.

- Collaborate closely with data scientists, ML engineers, and product teams to ensure robust and efficient deployment.

- Continuously improve deployment latency, inference throughput, and edge-device performance.

Requirements and Skills :

- Strong understanding of ML lifecycle and experience managing training-to-deployment pipelines.

- Hands-on experience with MLflow for model tracking and reproducibility.

- Proficient with Docker and container orchestration tools such as Kubernetes.

- Experience with CI/CD pipelines using tools like GitHub Actions, GitLab CI, or similar.

- Familiarity with TensorRT, ONNX, and model optimization for edge and embedded systems.

- Proven track record of deploying models on TFLite, CoreML, and WebAssembly platforms.

- Experience with workflow orchestration using Apache Airflow or similar tools.

- Exposure to monitoring tools like Prometheus, Grafana, Seldon, or WhyLabs.

- Strong scripting and automation skills in Python and Bash.

- Knowledge of biometric systems (face, speaker, fingerprint, palmprint, eye socket identification) is a strong plus.

- Excellent problem-solving skills, attention to detail, and ability to work in a fast-paced environment.

- Strong communication and documentation skills; experience collaborating in cross-functional teams.