Python Developer – AI Automation & Custom Model Specialist

Job Title

We are hiring an experienced Python Developer with deep expertise in AI automation, workflow orchestration, and custom model development to build next-generation healthcare applications. This role is ideal for developers who love training, modifying, and deploying production-grade AI systems while navigating the challenges of compliance, data governance, and real-time performance in healthcare environments.

Key Responsibilities:  AI Workflow Automation & Orchestration

• Build intelligent automation pipelines for clinical workflows
• Orchestrate real-time and batch AI workflows using Airflow, Prefect, or Dagster
• Develop event-driven architectures and human-in-the-loop validation layers
• Automate data ingestion, processing, and inference pipelines for medical data

Custom AI Model Development

• Train custom NLP, CV, or multi-modal models for medical tasks
• Fine-tune open-source models for domain-specific adaptation
• Use transfer learning on small datasets for clinical use
• Build ensemble learning systems for diagnostic accuracy

Open-Source Model Adaptation

• Modify models from Hugging Face, Meta, or Google for medical understanding
• Build custom tokenizers for EMR/EHR and medical terminology
• Apply quantization, pruning, and other inference optimizations
• Develop custom loss functions, training loops, and architectural variations

MLOps & Deployment

• Create training and CI/CD pipelines using MLflow, Kubeflow, or W&B
• Scale distributed training across multi-GPU environments (Ray/Horovod)
• Deploy models with FastAPI/Flask, supporting both batch and real-time inference
• Monitor for model drift, performance degradation, and compliance alerts

Healthcare AI Specialization

• Build systems for:
• Medical text classification & entity recognition
• Radiology report generation
• Clinical risk prediction
• Auto-coding & billing
• Real-time care alerts
• Ensure HIPAA compliance in all stages of model lifecycle

Required Qualifications Technical Skills

• 4–6 years of Python development
• 2+ years in ML/AI with deep learning frameworks (PyTorch/TensorFlow)
• Experience modifying open-source transformer models
• Strong expertise in workflow orchestration tools (Airflow, Prefect, Dagster)
• Hands-on with MLOps tools (MLflow, W&B, SageMaker, DVC)

Core Competencies

• Strong foundation in transformers, NLP, and CV
• Experience in distributed computing, GPU programming, and model compression
• Ability to explain and interpret model decisions (XAI, SHAP, LIME)
• Familiarity with containerized deployments (Docker, K8s)

 Technical Stack

• Languages: Python 3.9+, CUDA
• ML Frameworks: PyTorch, TensorFlow, Hugging Face, ONNX
• Workflow Tools: Airflow, Prefect, Dagster
• MLOps: MLflow, Weights & Biases, SageMaker
• Infra: AWS, Kubernetes, GPU Clusters
• Data: Spark, Dask, Pandas
• Databases: PostgreSQL, MongoDB, Delta Lake, S3
• Versioning: Git, DVC

 Preferred Qualifications

• Healthcare experience (EHR, medical NLP, radiology, DICOM, FHIR)
• Knowledge of federated learning, differential privacy, and AutoML
• Experience with multi-modal, multi-task, or edge model deployment
• Contributions to open-source projects, or research publications
• Knowledge of explainable AI and responsible ML practices

 Key Projects You’ll Work On

• Real-time clinical documentation automation
• Custom NER models for ICD/CPT tagging
• LLM adaptation for medical conversation understanding
• Real-time risk stratification pipelines for hospitals

What We Offer

• Comprehensive health, plans
• Flexible work options (remote/hybrid) with quarterly in-person meetups

Application Requirements

• Resume with ML/AI experience
• GitHub or portfolio links (model code, notebooks, demos)
• Cover letter describing your AI workflow or custom model build
• Code samples (open-source or private repos)
• Optional: Research papers, Kaggle profile, open challenges

Interview Process

• Initial HR screening (30 mins)
• Take-home Python + ML coding challenge
• Technical ML/AI deep-dive (90 mins)
• Model training/modification practical (2 hrs)
• System design for ML pipeline (60 mins)
• Presentation or walkthrough of past AI work (45 mins)
• Culture fit + final discussion
• References + offer

About the Role

You will help build AI that doesn’t just analyze data — it augments clinical decisions, automates medical documentation, and assists doctors in real-time. This role will impact thousands of patients and redefine how AI powers healthcare.

Aarna Tech Consultants Pvt. Ltd. (Atcuality) is an equal opportunity employer.

We believe in diversity, ethics, and inclusive AI systems for healthcare.