Software Engineer - AI Labs

United States Digital Space LLC; Maharashtra; On-site; INR 2,400,000 - 4,000,000; Full time

Job summary

United States Digital Space LLC is seeking a Software Engineer – AI Labs to join the ARMOR DSPM team in Mumbai. You will work with experienced AI/ML engineers to build production-grade AI solutions that discover, classify, and govern sensitive enterprise data across databases, file systems, SaaS apps, and endpoints.

The role emphasizes ML fundamentals, Python development, and scalable AI infrastructure. 2–5 years of hands-on experience in ML/NLP and software development is preferred, with an

Qualifications

• Bachelor's or Master's in CS/AI/ML or related field.
• 2–5 years of experience in ML, NLP, AI Engineering, or Software Development.
• Strong Python programming and software development fundamentals.
• Experience with ML workflows, evaluation metrics, and model deployment.

Responsibilities

• Build evaluation pipelines and measure model performance (Precision, Recall, F1).
• Design, develop, and optimize prompts for enterprise-scale classification tasks.
• Curate and maintain labeled datasets and improve data quality.
• Develop production-ready Python components for AI-powered classification pipelines.
• Support packaging, validation, and deployment of ML models across environments.

Skills

Machine Learning

NLP

AI Engineering

Software Development

Python

Git

CI/CD

Deployment

Education

Bachelor's or Master's degree in Computer Science, AI, ML, or related field

Tools

Hugging Face Transformers

LangChain

LangGraph

MLflow

RAGAS

Docker

Job description

Job Description Congratulations, you have taken the first step towards bagging a career-defining role. Join the team of superheroes that safeguard data wherever it goes.

What should you know about us?

We help the world’s leading enterprises share data fearlessly. Our data-centric approach to cybersecurity protects and controls information wherever it travels—inside or outside the organization. Enterprises choose the company to prevent data theft, safeguard critical digital assets, and stay compliant without sacrificing collaboration or productivity. We are embarking on an exciting and inspiring journey to expand our product footprint and grow our addressable market leveraging new capabilities offered by GenAI.

Know more about us at www.the company.com

You would love our tribe

If you are passionate about Artificial Intelligence, Machine Learning, and solving real-world cybersecurity challenges, this is the place for you! Join a team of innovators building next-generation AI capabilities that protect sensitive enterprise data across the globe.

A sneak peek into the role

the company is looking for a Software Engineer – AI Labs to join the ARMOR DSPM (Data Security Posture Management) team. You will work alongside experienced AI and Machine Learning engineers to build intelligent systems that discover, classify, and govern sensitive enterprise data across structured databases, unstructured file systems, SaaS applications, and endpoints.

This role offers an excellent opportunity for engineers with strong Machine Learning and LLM fundamentals who are eager to build production-grade AI solutions while contributing to one of the company's most strategic product initiatives.

Team: AI Labs – ARMOR Platform (DSPM)Location: MumbaiExperience: 2–5 YearsEmployment Type: Full-Time

Here’s what you will get to explore

AI Model Evaluation & Benchmarking

• Build and execute evaluation pipelines using ground truth datasets.
• Measure model performance using Precision, Recall, F1 Score, and other evaluation metrics.
• Benchmark model and prompt performance across multiple iterations.
• Continuously improve classifier accuracy through structured experimentation.

LLM Prompt Engineering

• Design, develop, and optimize prompts for enterprise-scale classification use cases.
• Perform regression testing to validate prompt quality and consistency.
• Treat prompts as reusable, version-controlled engineering assets.
• Experiment with structured outputs, few-shot prompting, and instruction tuning techniques.

Data Engineering & Classification

• Curate and maintain high-quality labeled datasets for training and evaluation.
• Work with annotation tools to improve data quality using analyst feedback.
• Develop and enhance components within the AI-powered classification pipeline.
• Write clean, scalable, and production-ready Python code.

AI Infrastructure & Deployment

• Support packaging, validation, and deployment of ML models across cloud, self-hosted, and air-gapped environments.
• Contribute to inference infrastructure and deployment automation.
• Collaborate with senior engineers to improve AI platform reliability and scalability.

Collaboration & Innovation

• Work closely with AI researchers, Product Engineering, and Security teams.
• Participate in technical discussions, architecture reviews, and experimentation initiatives.
• Contribute innovative ideas that enhance the AI capabilities of the ARMOR Platform.

We can see the next Entrepreneur at the company if you have

• Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• 2–3 years of experience in Machine Learning, NLP, AI Engineering, or Software Development.

Machine Learning & NLP

• Strong understanding of NLP concepts including tokenization, embeddings, Named Entity Recognition (NER), and text classification.
• Experience working with Transformer-based models using Hugging Face Transformers or equivalent frameworks.
• Hands-on knowledge of AI evaluation metrics including Precision, Recall, F1 Score, and Confusion Matrix.
• Exposure to evaluation frameworks such as RAGAS, DeepEval, MLflow, or similar tools.

LLM & Prompt Engineering

• Hands-on experience working with Large Language Models using OpenAI, Anthropic, or similar APIs.
• Experience developing prompts using structured outputs, few-shot learning, and instruction-based prompting.
• Working knowledge of LangChain, LangGraph, or equivalent LLM orchestration frameworks.
• Understanding of Retrieval-Augmented Generation (RAG), embeddings, and retrieval concepts.
• Curiosity to investigate prompt failures such as hallucinations, output drift, and instruction-following limitations.

Software Engineering

• Strong Python programming skills with experience writing clean, maintainable, and testable code.
• Familiarity with Git, CI/CD concepts, and collaborative software development practices.
• Ability to develop evaluation scripts, automate testing, and generate structured model evaluation reports.
• Basic understanding of Docker containers and deployment workflows.

Nice to Have

• Knowledge of Vector Databases and similarity search.
• Exposure to model fine-tuning or transfer learning techniques.
• Experience with OCR, Intelligent Document Processing (IDP), or document understanding.
• Understanding of data privacy and compliance standards such as GDPR, DPDP, or PCI DSS.
• Academic or professional experience in Information Extraction, Document Intelligence, or Applied NLP.

Why do we call Seclorites Entrepreneurs and not Employees?

• We empower people to think like owners, challenge conventions, and build innovative solutions.
• You will work on cutting-edge AI problems where experimentation, creativity, and continuous learning are encouraged.
• Receive direct mentorship from experienced AI leaders while working on real-world enterprise AI applications.
• Build AI solutions that protect sensitive information for Fortune 500 organizations across Banking, Healthcare, Defence, Government, and other highly regulated industries.
• At the company, your work has a direct impact on how organizations worldwide secure their most valuable data.

Excited to be the next Entrepreneur?

If you're passionate about Artificial Intelligence, Machine Learning, and solving meaningful cybersecurity challenges, we'd love to hear from you.