Software Engineer (AI/ML Focus)
Dimension1
2 - 5 years
Hyderabad
Posted: 04/04/2026
Job Description
Job Description: Software Engineer (AI/ML Focus)
Location: [Hyderabad / Remote]
Employment Type: Full-time
Application Link: https://forms.gle/yPxbhwBXxHh2vM8BA
About the Role
We are looking for a versatile Software Engineer with expertise in AI/ML who can build robust applications, design scalable systems, and integrate machine learning models into production environments. This role is ideal for someone who enjoys working across the full engineering stack from backend services and APIs to ML pipelines and deployment.
Key Responsibilities
Software Engineering
- Design, develop, and maintain high-quality backend services and APIs.
- Write clean, efficient, and testable code in Python, Java, or similar languages.
- Build scalable, cloud-native systems (AWS/GCP/Azure).
- Implement CI/CD pipelines, containerization (Docker/Kubernetes), and monitoring.
- Ensure security, reliability, and performance of applications.
AI/ML Engineering
- Collect, preprocess, and analyze data for training and evaluation.
- Build and fine-tune ML/DL models using PyTorch/TensorFlow/scikit-learn.
- Integrate ML models into production systems via APIs, microservices, or batch pipelines.
- Optimize models for performance, scalability, and latency in real-world applications.
- Monitor models in production for drift, reliability, and fairness.
Collaboration & Product Impact
- Work closely with product managers, designers, and data scientists to deliver end-to-end solutions.
- Translate business problems into engineering + ML solutions.
- Contribute to architectural decisions for both software systems and ML pipelines.
- Write clear documentation and mentor junior engineers where needed.
Qualifications
Must-have:
- Bachelors/Masters in Computer Science, Software Engineering, or related field.
- 36+ years of experience in software engineering
- hands-on ML experience.
- Strong programming skills in Python (preferred), plus Java/Go/Node for backend.
- Experience with ML frameworks (PyTorch, TensorFlow, scikit-learn).
- Solid understanding of algorithms, data structures, and system design
- Experience with databases (SQL + NoSQL), caching, and REST/GraphQL APIs.
- Knowledge of cloud services (AWS/GCP/Azure) and container orchestration (K8s).
Nice-to-have:
- Experience with MLOps (MLflow, Airflow, Kubeflow).
- Exposure to frontend frameworks (React, Angular, or Flutter for mobile).
- Knowledge of big data pipelines (Spark, Kafka, Dataflow).
- Prior work in NLP, Computer Vision, or Generative AI
Soft Skills
- Ability to work across the full stack of software + ML
- Strong problem-solving and debugging skills.
- Comfort working in fast-paced, startup-like environments.
- Passion for building real products that combine software and AI
What We Offer
- Competitive salary, equity, and benefits.
- Opportunity to work on cutting-edge AI systems in real-world products
- A role that blends software craftsmanship with AI innovation
- Career growth in both software engineering and AI/ML specialization
Services you might be interested in
Improve Your Resume Today
Boost your chances with professional resume services!
Get expert-reviewed, ATS-optimized resumes tailored for your experience level. Start your journey now.
