AI/ML Engineer

Neilsoft Research & Innovation Lab is building AI systems for engineering intelligence across domains such as BIM/CAD, digital twins, document intelligence, and conversational engineering assistants. We are looking for engineers who enjoy solving complex machine learning problems involving structured, visual, geometric, and textual data. The role involves working on computer vision for technical drawings, NLP for engineering documentation, geometric reasoning over 3D models, and AI copilots for engineering tools. You will design and deploy production-grade ML systems while contributing to the research and development of proprietary AI technologies.

Job Location : Pune

Position : AI/ML Engineer

Experience : 5 to 7 Years

Qualification : Bachelor / Master degree in: Computer Science / Engineering, Artificial Intelligence/Machine Learning, Data Science, Applied Mathematics.

You may work on one or more of the following areas:

1. Computer Vision for Engineering Drawings

Layout detection and element extraction from PDFs

Symbol detection and recognition

Floorplan understanding Diagram parsing

Object detection in technical drawings

2. Document Intelligence

Information extraction from engineering specifications

Table extraction from BOQ and cost documents

Entity extraction and semantic parsing

Document-level reasoning and cross-document linking

3. Geometry & 3D Intelligence

Graph-based reasoning on BIM models

Geometry classification and topology understanding

3D model feature extraction

Structural or spatial constraint detection

4. AI Copilot Systems for Engineering Tools

LLM-powered assistants

Natural language interfaces for BIM/CAD models

Retrieval-Augmented Generation (RAG) systems

Engineering knowledge systems

5. Multimodal AI Systems

Combining text, geometry, and visual data

Engineering knowledge graphs

Context-aware AI assistants

Key Responsibilities

Design and implement end-to-end ML systems from data ingestion to deployment.

Architect scalable pipelines for training, evaluation, and inference.

Lead development of models in: o Computer Vision o NLP o Multimodal learning

Design solutions for document understanding and geometric reasoning.

Implement RAG architectures for engineering knowledge systems.

Optimize models for latency, scalability, and production deployment.

Mentor junior engineers and guide ML experimentation.

Evaluate emerging ML techniques and integrate them into production systems.

Required Technical Skills & Experience

Advanced Machine Learning

Experience in:

Deep learning model development, Representation learning, Transformer architectures, Transfer learning, Multimodal ML systems

Computer Vision Experience with:

Object detection, Image segmentation, Document layout analysis, OCR pipelines

NLP / LLM Systems

Experience with:

Transformer-based models, Retrieval-Augmented Generation (RAG), Embedding models, Semantic search systems

MLOps

Experience building production ML systems using:

ML experiment tracking (MLflow / Weights & Biases), Model versioning, CI/CD pipelines, Monitoring and retraining pipelines

Data Systems

Vector databases, Data pipelines, Feature engineering systems

Preferred Expertise

Experience working with 3D data or geometric data

Knowledge of graph neural networks

Experience building AI copilots or AI assistants

Familiarity with BIM/CAD ecosystems

Experience deploying models on AWS / Azure / GCP

Tech Stack (Indicative) Candidates may work with:

Python, PyTorch, HuggingFace, OpenCV, FAISS / Vector DBs, FastAPI / microservices, Docker / Kubernetes, AWS / Azure ML platforms

Research Mindset

We value engineers who:

Read and implement latest ML research papers

Experiment with new model architectures

Build scalable ML systems

Combine research thinking with production engineering