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

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.

Salary: INR 15 to INR 30 lakhs per annum

Performance Bonus: Up to 10% of the base salary can be added

Location: Bangalore or Pune

Experience: 2-5 years

About AbleCredit:

AbleCredit is on a mission to solve the Credit Gap of emerging economies. In India alone, the Credit Gap is over USD 5T (Trillion!). This is the single largest contributor to poverty, poor genie index and lack of opportunities. Our Vision is to deploy AI reliably, and safely to solve some of the greatest problems of humanity.

Job Description:

This role is ideal for someone with a strong foundation in deep learning and hands-on experience with AI technologies.

• You will be tasked with solving complex, real-world problems using advanced machine learning models in a privacy-sensitive domain, where your contributions will have a direct impact on business-critical processes.
• As a Machine Learning Engineer at AbleCredit, you will collaborate closely with the founding team, who bring decades of industry expertise to the table.
• You’ll work on deploying cutting-edge Generative AI solutions at scale, ensuring they align with strict privacy requirements and optimize business outcomes.

This is an opportunity for experienced engineers to bring creative AI solutions to one of the most challenging and evolving sectors, while making a significant difference to the company’s growth and success.

Requirements:

• Experience: 2-4 years of hands-on experience in applying machine learning and deep learning techniques to solve complex business problems.
• Technical Skills: Proficiency in standard ML tools and languages, including:
• Python: Strong coding ability for building, training, and deploying machine learning models.
• PyTorch (or MLX or Jax): Solid experience in one or more deep learning frameworks for developing and fine-tuning models.
• Shell scripting: Familiarity with Unix/Linux shell scripting for automation and system-level tasks.
• Mathematical Foundation: Good understanding of the mathematical principles behind machine learning and deep learning (linear algebra, calculus, probability, optimization).
• Problem Solving: A passion for solving tough, ambiguous problems using AI, especially in data-sensitive, large-scale environments.
• Privacy & Security: Awareness and understanding of working in privacy-sensitive domains, adhering to best practices in data security and compliance.
• Collaboration: Ability to work closely with cross-functional teams, including engineers, product managers, and business stakeholders, and communicate technical ideas effectively.
• Work Experience: This position is for experienced candidates only.

Additional Information:

• Location: Pune or Bangalore
• Work Environment: Collaborative and entrepreneurial, with close interactions with the founders.
• Growth Opportunities: Exposure to large-scale AI systems, GenAI, and working in a data-driven privacy-sensitive domain.
• Compensation: Competitive salary and ESOPs, based on experience and performance
• Industry Impact: You’ll be at the forefront of applying Generative AI to solve high-impact problems in the finance/credit space, helping shape the future of AI in the business world.

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