10+ PyTorch Jobs in Pune | PyTorch Job openings in Pune

About NonStop io Technologies

NonStop io Technologies is a value-driven company with a strong focus on process-oriented software engineering. We specialize in Product Development and have a decade's worth of experience in building web and mobile applications across various domains. NonStop io Technologies follows core principles that guide its operations and believes in staying invested in a product's vision for the long term. We are a small but proud group of individuals who believe in the 'givers gain' philosophy and strive to provide value in order to seek value. We are committed to and specialize in building cutting-edge technology products and serving as trusted technology partners for startups and enterprises. We pride ourselves on fostering innovation, learning, and community engagement. Join us to work on impactful projects in a collaborative and vibrant environment.

Brief Description:

We're seeking an AI/ML Engineer to join our team. As AI/ML Engineer, you will be responsible for designing, developing, and implementing artificial intelligence (AI) and machine learning (ML) solutions to solve real-world business problems. You will work closely with engineering teams, including software engineers, domain experts, and product managers, to deploy and integrate Applied AI/ML solutions into the products that are being built at NonStop io. Your role will involve researching cutting-edge algorithms and data processing techniques, and implementing scalable solutions to drive innovation and improve the overall user experience.

Responsibilities

● Applied AI/ML engineering; Building engineering solutions on top of the AI/ML tooling available in the industry today. Eg: Engineering APIs around OpenAI

● AI/ML Model Development: Design, develop, and implement machine learning models and algorithms that address specific business challenges, such as natural language processing, computer vision, recommendation systems, anomaly detection, etc.

● Data Preprocessing and Feature Engineering: Cleanse, preprocess, and transform raw data into suitable formats for training and testing AI/ML models. Perform feature engineering to extract relevant features from the data

● Model Training and Evaluation: Train and validate AI/ML models using diverse datasets to achieve optimal performance. Employ appropriate evaluation metrics to assess model accuracy, precision, recall, and other relevant metrics

● Data Visualization: Create clear and insightful data visualizations to aid in understanding data patterns, model behaviour, and performance metrics

● Deployment and Integration: Collaborate with software engineers and DevOps teams to deploy AI/ML models into production environments and integrate them into various applications and systems

● Data Security and Privacy: Ensure compliance with data privacy regulations and implement security measures to protect sensitive information used in AI/ML processes

● Continuous Learning: Stay updated with the latest advancements in AI/ML research, tools, and technologies, and apply them to improve existing models and develop novel solutions

● Documentation: Maintain detailed documentation of the AI/ML development process, including code, models, algorithms, and methodologies for easy understanding and future reference.

Qualifications & Skills

● Bachelor's, Master's, or PhD in Computer Science, Data Science, Machine Learning, or a related field. Advanced degrees or certifications in AI/ML are a plus

● Proven experience as an AI/ML Engineer, Data Scientist, or related role, ideally with a strong portfolio of AI/ML projects

● Proficiency in programming languages commonly used for AI/ML. Preferably Python

● Familiarity with popular AI/ML libraries and frameworks, such as TensorFlow, PyTorch, scikit-learn, etc.

● Familiarity with popular AI/ML Models such as GPT3, GPT4, Llama2, BERT etc.

● Strong understanding of machine learning algorithms, statistics, and data structures

● Experience with data preprocessing, data wrangling, and feature engineering

● Knowledge of deep learning architectures, neural networks, and transfer learning

● Familiarity with cloud platforms and services (e.g., AWS, Azure, Google Cloud) for scalable AI/ML deployment

● Solid understanding of software engineering principles and best practices for writing maintainable and scalable code

● Excellent analytical and problem-solving skills, with the ability to think critically and propose innovative solutions

● Effective communication skills to collaborate with cross-functional teams and present complex technical concepts to non-technical stakeholders

What You’ll Do:

As a Sr. Data Scientist, you will work closely across DeepIntent Data Science teams located in New York, India, and Bosnia. The role will focus on building predictive models, implementing data-driven solutions to maximize ad effectiveness. You will also lead efforts in generating analyses and insights related to the measurement of campaign outcomes, Rx, patient journey, and supporting the evolution of the DeepIntent product suite. Activities in this position include developing and deploying models in production, reading campaign results, analyzing medical claims, clinical, demographic and clickstream data, performing analysis and creating actionable insights, summarizing, and presenting results and recommended actions to internal stakeholders and external clients, as needed.

• Explore ways to create better predictive models.
• Analyze medical claims, clinical, demographic and clickstream data to produce and present actionable insights.
• Explore ways of using inference, statistical, and machine learning techniques to improve the performance of existing algorithms and decision heuristics.
• Design and deploy new iterations of production-level code.
• Contribute posts to our upcoming technical blog.

Who You Are:

• Bachelor’s degree in a STEM field, such as Statistics, Mathematics, Engineering, Biostatistics, Econometrics, Economics, Finance, or Data Science.
• 5+ years of working experience as a Data Scientist or Researcher in digital marketing, consumer advertisement, telecom, or other areas requiring customer-level predictive analytics.
• Advanced proficiency in performing statistical analysis in Python, including relevant libraries, is required.
• Experience working with data processing, transformation and building model pipelines using tools such as Spark, Airflow, and Docker.
• You have an understanding of the ad-tech ecosystem, digital marketing and advertising data and campaigns or familiarity with the US healthcare patient and provider systems (e.g. medical claims, medications).
• You have varied and hands-on predictive machine learning experience (deep learning, boosting algorithms, inference…).
• You are interested in translating complex quantitative results into meaningful findings and interpretable deliverables, and communicating with less technical audiences orally and in writing.
• You can write production level code, work with Git repositories.
• Active Kaggle participant.
• Working experience with SQL.
• Familiar with medical and healthcare data (medical claims, Rx, preferred).
• Conversant with cloud technologies such as AWS or Google Cloud.

Position: Machine Learning Engineer 

Job Type: Full Time, Permanent 

Location : Baner, Pune 

We are looking for a visionary Machine Learning Engineer to bridge the gap between cutting-edge research and real-world applications. In this role, you won’t just be training models; you will be architecting the visual intelligence that powers our next generation of products. You will join a high-impact team dedicated to solving complex spatial and visual problems at scale. 

What You’ll Do 

• Architect & Innovate: Own the journey from research to production-ready computer vision pipelines. 
• Optimize Performance: Build high-speed deep learning models for real-time detection and recognition. 
• Lead with Data: Curate massive datasets and apply advanced engineering to maximize model accuracy. 
• Collaborate & Integrate: Partner with cross-functional teams to embed AI insights into core products. 
• Stay Ahead: Proactively prototype SOTA architectures to keep our solutions at the cutting edge. 
• 
  

What We’re Looking For 

• Expert in Python and OpenCV. You are framework-agnostic but a power user of PyTorch or TensorFlow. 
• 2–5 years of shipping ML products. You know how to move beyond research notebooks into scalable production code. 
• A problem solver obsessed with robust, explainable AI and clean, rigorous validation. 
• Strong mastery of the math behind CV—specifically geometry, linear algebra, and optimization. 

Role: Data Scientist (Python + R Expertise)

Exp: 8 -12 Years

CTC: up to 30 LPA

Required Skills & Qualifications:

• 8–12 years of hands-on experience as a Data Scientist or in a similar analytical role.
• Strong expertise in Python and R for data analysis, modeling, and visualization.
• Proficiency in machine learning frameworks (scikit-learn, TensorFlow, PyTorch, caret, etc.).
• Strong understanding of statistical modeling, hypothesis testing, regression, and classification techniques.
• Experience with SQL and working with large-scale structured and unstructured data.
• Familiarity with cloud platforms (AWS, Azure, or GCP) and deployment practices (Docker, MLflow).
• Excellent analytical, problem-solving, and communication skills.

Preferred Skills:

• Experience with NLP, time series forecasting, or deep learning projects.
• Exposure to data visualization tools (Tableau, Power BI, or R Shiny).
• Experience working in product or data-driven organizations.
• Knowledge of MLOps and model lifecycle management is a plus.

If interested kindly share your updated resume on 82008 31681

JioTesseract, a digital arm of Reliance Industries, is India's leading and largest AR/VR organization with the mission to democratize mixed reality for India and the world. We make products at the cross of hardware, software, content and services with focus on making India the leader in spatial computing. We specialize in creating solutions in AR, VR and AI, with some of our notable products such as JioGlass, JioDive, 360 Streaming, Metaverse, AR/VR headsets for consumers and enterprise space.

Mon-fri role, In office, with excellent perks and benefits!

Position Overview

We are seeking a Software Architect to lead the design and development of high-performance robotics and AI software stacks utilizing NVIDIA technologies. This role will focus on defining scalable, modular, and efficient architectures for robot perception, planning, simulation, and embedded AI applications. You will collaborate with cross-functional teams to build next-generation autonomous systems 9

Key Responsibilities:

1. System Architecture & Design

● Define scalable software architectures for robotics perception, navigation, and AI-driven decision-making.

● Design modular and reusable frameworks that leverage NVIDIA’s Jetson, Isaac ROS, Omniverse, and CUDA ecosystems.

● Establish best practices for real-time computing, GPU acceleration, and edge AI inference.

2. Perception & AI Integration

● Architect sensor fusion pipelines using LIDAR, cameras, IMUs, and radar with DeepStream, TensorRT, and ROS2.

● Optimize computer vision, SLAM, and deep learning models for edge deployment on Jetson Orin and Xavier.

● Ensure efficient GPU-accelerated AI inference for real-time robotics applications.

3. Embedded & Real-Time Systems

● Design high-performance embedded software stacks for real-time robotic control and autonomy.

● Utilize NVIDIA CUDA, cuDNN, and TensorRT to accelerate AI model execution on Jetson platforms.

● Develop robust middleware frameworks to support real-time robotics applications in ROS2 and Isaac SDK.

4. Robotics Simulation & Digital Twins

● Define architectures for robotic simulation environments using NVIDIA Isaac Sim & Omniverse.

● Leverage synthetic data generation (Omniverse Replicator) for training AI models.

● Optimize sim-to-real transfer learning for AI-driven robotic behaviors.

5. Navigation & Motion Planning

● Architect GPU-accelerated motion planning and SLAM pipelines for autonomous robots.

● Optimize path planning, localization, and multi-agent coordination using Isaac ROS Navigation.

● Implement reinforcement learning-based policies using Isaac Gym.

6. Performance Optimization & Scalability

● Ensure low-latency AI inference and real-time execution of robotics applications.

● Optimize CUDA kernels and parallel processing pipelines for NVIDIA hardware.

● Develop benchmarking and profiling tools to measure software performance on edge AI devices.

Required Qualifications:

● Master’s or Ph.D. in Computer Science, Robotics, AI, or Embedded Systems.

● Extensive experience (7+ years) in software development, with at least 3-5 years focused on architecture and system design, especially for robotics or embedded systems.

● Expertise in CUDA, TensorRT, DeepStream, PyTorch, TensorFlow, and ROS2.

● Experience in NVIDIA Jetson platforms, Isaac SDK, and GPU-accelerated AI.

● Proficiency in programming languages such as C++, Python, or similar, with deep understanding of low-level and high-level design principles.

● Strong background in robotic perception, planning, and real-time control.

● Experience with cloud-edge AI deployment and scalable architectures.

Preferred Qualifications

● Hands-on experience with NVIDIA DRIVE, NVIDIA Omniverse, and Isaac Gym

● Knowledge of robot kinematics, control systems, and reinforcement learning

● Expertise in distributed computing, containerization (Docker), and cloud robotics

● Familiarity with automotive, industrial automation, or warehouse robotics

● Experience designing architectures for autonomous systems or multi-robot systems.

● Familiarity with cloud-based solutions, edge computing, or distributed computing for robotics

● Experience with microservices or service-oriented architecture (SOA)

● Knowledge of machine learning and AI integration within robotic systems

● Knowledge of testing on edge devices with HIL and simulations (Isaac Sim, Gazebo, V-REP etc.)

JioTesseract, a digital arm of Reliance Industries, is India's leading and largest AR/VR organization with the mission to democratize mixed reality for India and the world. We make products at the cross of hardware, software, content and services with focus on making India the leader in spatial computing. We specialize in creating solutions in AR, VR and AI, with some of our notable products such as JioGlass, JioDive, 360 Streaming, Metaverse, AR/VR headsets for consumers and enterprise space.

Mon-Fri, In office role with excellent perks and benefits!

Key Responsibilities:

1. Design, develop, and maintain backend services and APIs using Node.js or Python, or Java.

2. Build and implement scalable and robust microservices and integrate API gateways.

3. Develop and optimize NoSQL database structures and queries (e.g., MongoDB, DynamoDB).

4. Implement real-time data pipelines using Kafka.

5. Collaborate with front-end developers to ensure seamless integration of backend services.

6. Write clean, reusable, and efficient code following best practices, including design patterns.

7. Troubleshoot, debug, and enhance existing systems for improved performance.

Mandatory Skills:

1. Proficiency in at least one backend technology: Node.js or Python, or Java.

2. Strong experience in:

i. Microservices architecture,

ii. API gateways,

iii. NoSQL databases (e.g., MongoDB, DynamoDB),

iv. Kafka

v. Data structures (e.g., arrays, linked lists, trees).

3. Frameworks:

i. If Java : Spring framework for backend development.

ii. If Python: FastAPI/Django frameworks for AI applications.

iii. If Node: Express.js for Node.js development.

Good to Have Skills:

1. Experience with Kubernetes for container orchestration.

2. Familiarity with in-memory databases like Redis or Memcached.

3. Frontend skills: Basic knowledge of HTML, CSS, JavaScript, or frameworks like React.js.

We are seeking a Senior Data Scientist with hands-on experience in Generative AI (GenAI) and Large Language Models (LLM). The ideal candidate will have expertise in building, fine-tuning, and deploying LLMs, as well as managing the lifecycle of AI models through LLMOps practices. You will play a key role in driving AI innovation, developing advanced algorithms, and optimizing model performance for various business applications.

Key Responsibilities:

• Develop, fine-tune, and deploy Large Language Models (LLM) for various business use cases.
• Implement and manage the operationalization of LLMs using LLMOps best practices.
• Collaborate with cross-functional teams to integrate AI models into production environments.
• Optimize and troubleshoot model performance to ensure high accuracy and scalability.
• Stay updated with the latest advancements in Generative AI and LLM technologies.

Required Skills and Qualifications:

• Strong hands-on experience with Generative AI, LLMs, and NLP techniques.
• Proven expertise in LLMOps, including model deployment, monitoring, and maintenance.
• Proficiency in programming languages like Python and frameworks such as TensorFlow, PyTorch, or Hugging Face.
• Solid understanding of AI/ML algorithms and model optimization.

Who are we looking for?  

We are looking for a Senior Data Scientist, who will design and develop data-driven solutions using state-of-the-art methods. You should be someone with strong and proven experience in working on data-driven solutions. If you feel you’re enthusiastic about transforming business requirements into insightful data-driven solutions, you are welcome to join our fast-growing team to unlock your best potential.  

Job Summary 

• Supporting company mission by understanding complex business problems through data-driven solutions. 
• Designing and developing machine learning pipelines in Python and deploying them in AWS/GCP, ... 
• Developing end-to-end ML production-ready solutions and visualizations. 
• Analyse large sets of time-series industrial data from various sources, such as production systems, sensors, and databases to draw actionable insights and present them via custom dashboards. 
• Communicating complex technical concepts and findings to non-technical stakeholders of the projects 
• Implementing the prototypes using suitable statistical tools and artificial intelligence algorithms. 
• Preparing high-quality research papers and participating in conferences to present and report experimental results and research findings. 
• Carrying out research collaborating with internal and external teams and facilitating review of ML systems for innovative ideas to prototype new models. 

Qualification and experience 

• B.Tech/Masters/Ph.D. in computer science, electrical engineering, mathematics, data science, and related fields. 
• 5+ years of professional experience in the field of machine learning, and data science. 
• Experience with large-scale Time-series data-based production code development is a plus. 

Skills and competencies 

• Familiarity with Docker, and ML Libraries like PyTorch, sklearn, pandas, SQL, and Git is a must. 
• Ability to work on multiple projects. Must have strong design and implementation skills. 
• Ability to conduct research based on complex business problems. 
• Strong presentation skills and the ability to collaborate in a multi-disciplinary team. 
• Must have programming experience in Python. 
• Excellent English communication skills, both written and verbal. 

Benefits and Perks

• Culture of innovation, creativity, learning, and even failure, we believe in bringing out the best in you. 
• Progressive leave policy for effective work-life balance. 
• Get mentored by highly qualified internal resource groups and opportunity to avail industry-driven mentorship program, as we believe in empowering people.  
• Multicultural peer groups and supportive workplace policies.  
• Work from beaches, hills, mountains, and many more with the yearly workcation program; we believe in mixing elements of vacation and work. 

 Hiring Process 

• Call with Talent Acquisition Team: After application screening, a first-level screening with the talent acquisition team to understand the candidate's goals and alignment with the job requirements. 
• First Round: Technical round 1 to gauge your domain knowledge and functional expertise. 
• Second Round: In-depth technical round and discussion about the departmental goals, your role, and expectations.
• Final HR Round: Culture fit round and compensation discussions.
• Offer: Congratulations you made it!  

If this position sparked your interest, apply now to initiate the screening process.

About UpSolve

We built and deliver complex AI solutions which help drive business decisions faster and more accurately. We are a typical AI company and have a range of solutions developed on Video, Image and Text.

What you will do

• Stay informed on new technologies and implement cautiously
• Maintain necessary documentation for the project
• Fix the issues reported by application users
• Plan, build, and design solutions with a mental note of future requirements
• Coordinate with the development team to manage fixes, code changes, and merging

Location: Mumbai

Working Mode: Remote

What are we looking for

• Bachelor's or Master's degree in Computer Science, Software Engineering, or a related field.
• Minimum 2 years of professional experience in software development, with a focus on machine learning and full stack development.
• Strong proficiency in Python programming language and its machine learning libraries such as TensorFlow, PyTorch, or scikit-learn.
• Experience in developing and deploying machine learning models in production environments.
• Proficiency in web development technologies including HTML, CSS, JavaScript, and front-end frameworks such as React, Angular, or Vue.js.
• Experience in designing and developing RESTful APIs and backend services using frameworks like Flask or Django.
• Knowledge of databases and SQL for data storage and retrieval.
• Familiarity with version control systems such as Git.
• Strong problem-solving and analytical skills.
• Excellent communication and collaboration abilities.
• Ability to work effectively in a fast-paced and dynamic team environment.
• Good to have Cloud Exposure