PyTorch & Python Developer

Job Description

We are seeking a skilled PyTorch & Python Developer to work on AI and machine learning-based projects. In this role, you will design, implement, and optimize deep learning models for various applications. You will collaborate with data scientists and software engineers to create scalable and efficient machine learning solutions. The ideal candidate should have hands-on experience with PyTorch and Python, and a good understanding of machine learning algorithms and frameworks.

Key Responsibilities:**

– Develop and implement machine learning models using **PyTorch** and **Python**.

– Work on end-to-end model development, including data preprocessing, feature engineering, model training, and evaluation.

– Optimize and fine-tune machine learning models to improve performance and efficiency.

– Collaborate with cross-functional teams to integrate machine learning models into production systems.

– Write clean, maintainable, and efficient code to support machine learning pipelines.

– Assist in model deployment and scaling for production environments.

– Stay up-to-date with the latest advancements in machine learning, AI, and deep learning technologies.

– Write unit tests and document code to ensure the quality of the work.

– Conduct experiments, analyze results, and provide insights based on data.

– Work closely with data engineers to build scalable data pipelines for machine learning tasks.
*Required Skills & Qualifications:**

– **2-3 years** of professional experience in Python programming, with a focus on machine learning and AI development.

– Hands-on experience with **PyTorch** for developing and deploying deep learning models.

– Strong understanding of **machine learning** concepts and algorithms, including supervised and unsupervised learning, classification, regression, and neural networks.

– Proficiency in **Python** libraries such as NumPy, Pandas, and Matplotlib for data analysis and visualization.

– Familiarity with version control tools like **Git** for collaborative development.

– Experience with model evaluation techniques, hyperparameter tuning, and model optimization.

– Understanding of **data structures**, **algorithms**, and software development best practices.

– Familiarity with cloud platforms like **AWS**, **Azure**, or **Google Cloud** for deploying models is a plus.

– Good problem-solving skills and the ability to work independently as well as part of a team.

– Strong communication skills and the ability to present technical findings to non-technical stakeholders.
*Preferred Skills (Optional):**

– Experience with other machine learning frameworks such as **TensorFlow** or **Keras**.

– Knowledge of **Docker** and containerization for model deployment.

– Experience with **NLP** (Natural Language Processing) or **Computer Vision** applications.

– Familiarity with **data engineering** principles and tools like Apache Spark or Hadoop.
Education & Experience:**

– Bachelors or Masters degree in **Computer Science**, **Data Science**, **Engineering**, or a related field.

– 2-3 years of hands-on experience in Python and PyTorch development.

Job Types: Full-time, Permanent

Benefits:
Work from home

Location Type:
Remote

Schedule:
Day shift
Monday to Friday

Education:
Bachelor’s (Preferred)

Experience:
Python: 2 years (Preferred)
total work: 2 years (Preferred)

Work Location: Remote,