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

CoinFantasy is looking for an experienced Senior AI Architect to lead both the decentralised protocol development and the design of AI-driven applications on this network. As a visionary in AI and distributed computing, you will play a central role in shaping the protocol’s technical direction, enabling efficient task distribution, and scaling AI use cases across a heterogeneous, decentralised infrastructure.

Job Responsibilities

• Architect and oversee the protocol’s development, focusing on dynamic node orchestration, layer-wise model sharding, and secure, P2P network communication.
• Drive the end-to-end creation of AI applications, ensuring they are optimised for decentralised deployment and include use cases with autonomous agent workflows.
• Architect AI systems capable of running on decentralised networks, ensuring they balance speed, scalability, and resource usage.
• Design data pipelines and governance strategies for securely handling large-scale, decentralised datasets.
• Implement and refine strategies for swarm intelligence-based task distribution and resource allocation across nodes. Identify and incorporate trends in decentralised AI, such as federated learning and swarm intelligence, relevant to various industry applications.
• Lead cross-functional teams in delivering full-precision computing and building a secure, robust decentralised network.
• Represent the organisation’s technical direction, serving as the face of the company at industry events and client meetings.

Requirements

• Bachelor’s/Master’s/Ph.D. in Computer Science, AI, or related field.
• 12+ years of experience in AI/ML, with a track record of building distributed systems and AI solutions at scale.
• Strong proficiency in Python, Golang, and machine learning frameworks (e.g., TensorFlow, PyTorch).
• Expertise in decentralised architecture, P2P networking, and heterogeneous computing environments.
• Excellent leadership skills, with experience in cross-functional team management and strategic decision-making.
• Strong communication skills, adept at presenting complex technical solutions to diverse audiences.

About Us

CoinFantasy is a Play to Invest platform that brings the world of investment to users through engaging games. With multiple categories of games, it aims to make investing fun, intuitive, and enjoyable for users. It features a sandbox environment in which users are exposed to the end-to-end investment journey without risking financial losses.

Building on this foundation, we are now developing a groundbreaking decentralised protocol that will transform the AI landscape.

Website:

Benefits

• Competitive Salary
• An opportunity to be part of the Core team in a fast-growing company
• A fulfilling, challenging and flexible work experience
• Practically unlimited professional and career growth opportunities
• 
  

🎯 Ideal Candidate Profile:

This role requires a seasoned engineer/scientist with a strong academic background from a premier institution and significant hands-on experience in deep learning (specifically image processing) within a hardware or product manufacturing environment.

📋 Must-Have Requirements:

Experience & Education Combinations:

Candidates must meet one of the following criteria:

• Doctorate (PhD) + 2 years of related work experience
• Master's Degree + 5 years of related work experience
• Bachelor's Degree + 7 years of related work experience

Technical Skills:

• Minimum 5 years of hands-on experience in all of the following:
• Python
• Deep Learning (DL)
• Machine Learning (ML)
• Algorithm Development
• Image Processing
• 3.5 to 4 years of strong proficiency with PyTorch OR TensorFlow / Keras.

Industry & Institute:

• Education: Must be from a premier institute (IIT, IISC, IIIT, NIT, BITS) or a recognized regional tier 1 college.
• Industry: Current or past experience in a Product, Semiconductor, or Hardware Manufacturing company is mandatory.
• Preference: Candidates from engineering product companies are strongly preferred.

ℹ️ Additional Role Details:

• Interview Process: 3 technical rounds followed by 1 HR round.
• Work Model: Hybrid (requiring 3 days per week in the office).

Based on the job description you provided, here is a detailed breakdown of the Required Skills and Qualifications for this AI/ML/LLM role, formatted for clarity.

📝 Required Skills and Competencies:

💻 Programming & ML Prototyping:

• Strong Proficiency: Python, Data Structures, and Algorithms.
• Hands-on Experience: NumPy, Pandas, Scikit-learn (for ML prototyping).

🤖 Machine Learning Frameworks:

• Core Concepts: Solid understanding of:
• Supervised/Unsupervised Learning
• Regularization
• Feature Engineering
• Model Selection
• Cross-Validation
• Ensemble Methods: Experience with models like XGBoost and LightGBM.

🧠 Deep Learning Techniques:

• Frameworks: Proficiency with PyTorch OR TensorFlow / Keras.
• Architectures: Knowledge of:
• Convolutional Neural Networks (CNNs)
• Recurrent Neural Networks (RNNs)
• Long Short-Term Memory networks (LSTMs)
• Transformers
• Attention Mechanisms
• Optimization: Familiarity with optimization techniques (e.g., Adam, SGD), Dropout, and Batch Normalization.

💬 LLMs & RAG (Retrieval-Augmented Generation):

• Hugging Face: Experience with the Transformers library (tokenizers, embeddings, model fine-tuning).
• Vector Databases: Familiarity with Milvus, FAISS, Pinecone, or ElasticSearch.
• Advanced Techniques: Proficiency in:
• Prompt Engineering
• Function/Tool Calling
• JSON Schema Outputs

🛠️ Data & Tools:

• Data Management: SQL fundamentals; exposure to data wrangling and pipelines.
• Tools: Experience with Git/GitHub, Jupyter, and basic Docker.

🎓 Minimum Qualifications (Experience & Education Combinations):

Candidates must have experience building AI systems/solutions with Machine Learning, Deep Learning, and LLMs, meeting one of the following criteria:

• Doctorate (Academic) Degree + 2 years of related work experience.
• Master's Level Degree + 5 years of related work experience.
• Bachelor's Level Degree + 7 years of related work experience.

⭐ Preferred Traits and Mindset:

• Academic Foundation: Solid academic background with strong applied ML/DL exposure.
• Curiosity: Eagerness to learn cutting-edge AI and willingness to experiment.
• Communication: Clear communicator who can explain ML/LLM trade-offs simply.
• Ownership: Strong problem-solving and ownership mindset.

JOB DESCRIPTION/PREFERRED QUALIFICATIONS:

REQUIRED SKILLS/COMPETENCIES:

Programming Languages:

• Strong in Python, data structures, and algorithms.
• Hands-on with NumPy, Pandas, Scikit-learn for ML prototyping.

Machine Learning Frameworks:

• Understanding of supervised/unsupervised learning, regularization, feature engineering, model selection, cross-validation, ensemble methods (XGBoost, LightGBM).

Deep Learning Techniques:

• Proficiency with PyTorch or TensorFlow/Keras
• Knowledge of CNNs, RNNs, LSTMs, Transformers, Attention mechanisms.
• Familiarity with optimization (Adam, SGD), dropout, batch norm.

LLMs & RAG:

• Hugging Face Transformers (tokenizers, embeddings, model fine-tuning).
• Vector databases (Milvus, FAISS, Pinecone, ElasticSearch).
• Prompt engineering, function/tool calling, JSON schema outputs.

Data & Tools:

• SQL fundamentals; exposure to data wrangling and pipelines.
• Git/GitHub, Jupyter, basic Docker.

WHAT ARE WE LOOKING FOR?

• Solid academic foundation with strong applied ML/DL exposure.
• Curiosity to learn cutting-edge AI and willingness to experiment.
• Clear communicator who can explain ML/LLM trade-offs simply.
• Strong problem-solving and ownership mindset.

MINIMUM QUALIFICATIONS:

• Doctorate (Academic) Degree and 2 years related work experience; Master's Level Degree and related work experience of 5 years; Bachelor's Level Degree and related work experience of 7 years in building AI systems/solutions with Machine Learning, Deep Learning, and LLMs.

MUST-HAVES:

• Education/qualification:  Preferably from premier Institute like IIT, IISC, IIIT, NIT and BITS. Also regional tier 1 colleges.

• Doctorate (Academic) Degree and 2 years related work experience; or Master's Level Degree and related work experience of 5 years; or Bachelor's Level Degree and related work experience of 7 years

• Min 5 yrs experience in the Mandatory Skills: Python, Deep Learning, Machine Learning, Algorithm Development and Image Processing

• 3.5 to 4 yrs proficiency with PyTorch or TensorFlow/Keras

• Candidates from engineering product companies have higher chances of getting shortlisted (current company or past experience)

QUESTIONNAIRE: 

Do you have at least 5 years of experience with Python, Deep Learning, Machine Learning, Algorithm Development, and Image Processing? Please mention the skills and years of experience:

Do you have experience with PyTorch or TensorFlow / Keras?

• PyTorch
• TensorFlow / Keras
• Both

How many years of experience do you have with PyTorch or TensorFlow / Keras?

• Less than 3 years
• 3 to 3.5 years
• 3.5 to 4 years
• More than 4 years

Is the candidate willing to relocate to Chennai?

• Ready to relocate
• Based in Chennai

What type of company have you worked for in your career?

• Service-based IT company
• Product company
• Semiconductor company
• Hardware manufacturing company
• None of the above

JOB DESCRIPTION/PREFERRED QUALIFICATIONS:

KEY RESPONSIBILITIES: 

• Lead and mentor a team of algorithm engineers, providing guidance and support to ensure their professional growth and success. 
• Develop and maintain the infrastructure required for the deployment and execution of algorithms at scale. 
• Collaborate with data scientists, software engineers, and product managers to design and implement robust and scalable algorithmic solutions. 
• Optimize algorithm performance and resource utilization to meet business objectives.
• Stay up to date with the latest advancements in algorithm engineering and infrastructure technologies and apply them to improve our systems.
• Drive continuous improvement in development processes, tools, and methodologies. 

QUALIFICATIONS: 

• Bachelor's or Master's degree in Computer Science, Engineering, or a related field. 
• Proven experience in developing computer vision and image processing algorithms and ML/DL algorithms. 
• Familiar with high performance computing, parallel programming and distributed systems.
• Strong leadership and team management skills, with a track record of successfully leading engineering teams. 
• Proficiency in programming languages such as Python, C++ and CUDA. 
• Excellent problem-solving and analytical skills. 
• Strong communication and collaboration abilities. 

PREFERRED QUALIFICATIONS: 

• Experience with machine learning frameworks and libraries (e.g., TensorFlow, PyTorch, Scikit-learn). 
• Experience with GPU architecture and algo development toolkits like Docker, Apptainer. 

MINIMUM QUALIFICATIONS: 

• Bachelor's degree plus 8 + years of experience 
• Master's degree plus 8 + years of experience 
• Familiar with high performance computing, parallel programming and distributed systems.

MUST-HAVE SKILLS: 

• Phd with 6 yrs industry exp or M.Tech + 8 yrs experience or B.Tech + 10 yrs experience.
• 14 yrs exp if an IC role.
• Minimum 1 yrs experience working as a Manager/Lead
• 8 years' experience in any of the programming languages such as Python/C++/CUDA.
• 8 years' experience in Machine learning, Artificial intelligence, Deep learning.
• 2 to 3 years exp in Image processing & Computer vision is a MUST
• Product / Semi-conductor / Hardware Manufacturing company experience is a MUST. Candidates should be from engineering product companies 
• Candidates from Tier 1 colleges like (IIT, IIIT, VIT, NIT) (Preferred)
• Relocation to Chennai is mandatory

NICE TO HAVE SKILLS: 

• Candidates from Semicon or manufacturing companies
• Candidates with more than 8 CPGA

Job Title : Senior Machine Learning Engineer

Experience : 8+ Years

Location : Chennai

Notice Period : Immediate Joiners Only

Work Mode : Hybrid

Job Summary :

We are seeking an experienced Machine Learning Engineer with a strong background in Python, ML algorithms, and data-driven development.

The ideal candidate should have hands-on experience with popular ML frameworks and tools, solid understanding of clustering and classification techniques, and be comfortable working in Unix-based environments with Agile teams.

Mandatory Skills :

• Programming Languages : Python
• Machine Learning : Strong experience with ML algorithms, models, and libraries such as Scikit-learn, TensorFlow, and PyTorch
• ML Concepts : Proficiency in supervised and unsupervised learning, including techniques such as K-Means, DBSCAN, and Fuzzy Clustering
• Operating Systems : RHEL or any Unix-based OS
• Databases : Oracle or any relational database
• Version Control : Git
• Development Methodologies : Agile

Desired Skills :

• Experience with issue tracking tools such as Azure DevOps or JIRA.
• Understanding of data science concepts.
• Familiarity with Big Data algorithms, models, and libraries.

Role Overview:

We are seeking a highly skilled and motivated Data Scientist to join our growing team. The ideal candidate will be responsible for developing and deploying machine learning models from scratch to production level, focusing on building robust data-driven products. You will work closely with software engineers, product managers, and other stakeholders to ensure our AI-driven solutions meet the needs of our users and align with the company's strategic goals.

Key Responsibilities:

• Develop, implement, and optimize machine learning models and algorithms to support product development.
• Work on the end-to-end lifecycle of data science projects, including data collection, preprocessing, model training, evaluation, and deployment.
• Collaborate with cross-functional teams to define data requirements and product taxonomy.
• Design and build scalable data pipelines and systems to support real-time data processing and analysis.
• Ensure the accuracy and quality of data used for modeling and analytics.
• Monitor and evaluate the performance of deployed models, making necessary adjustments to maintain optimal results.
• Implement best practices for data governance, privacy, and security.
• Document processes, methodologies, and technical solutions to maintain transparency and reproducibility.

Qualifications:

• Bachelor's or Master's degree in Data Science, Computer Science, Engineering, or a related field.
• 5+ years of experience in data science, machine learning, or a related field, with a track record of developing and deploying products from scratch to production.
• Strong programming skills in Python and experience with data analysis and machine learning libraries (e.g., Pandas, NumPy, TensorFlow, PyTorch).
• Experience with cloud platforms (e.g., AWS, GCP, Azure) and containerization technologies (e.g., Docker).
• Proficiency in building and optimizing data pipelines, ETL processes, and data storage solutions.
• Hands-on experience with data visualization tools and techniques.
• Strong understanding of statistics, data analysis, and machine learning concepts.
• Excellent problem-solving skills and attention to detail.
• Ability to work collaboratively in a fast-paced, dynamic environment.

Preferred Qualifications:

• Knowledge of microservices architecture and RESTful APIs.
• Familiarity with Agile development methodologies.
• Experience in building taxonomy for data products.
• Strong communication skills and the ability to explain complex technical concepts to non-technical stakeholders.