Senior Machine Learning Engineer Role (Python)

Responsibilities:

Algorithm Development
  • Design, implement, and optimize advanced machine learning algorithms for ML development based on PyTorch or TensorFlow.

  • Collaborate with cross-functional teams to understand business requirements and translate them into scalable and efficient ML models.

Model Training and Evaluation
  • Train, fine-tune, and evaluate machine learning models using large datasets.

  • Implement and adhere to best practices for model validation, testing, and deployment.

Feature Engineering
  • Conduct comprehensive labeling, and feature engineering to enhance the performance of ML models.

  • Collaborate closely with data scientists and domain experts to identify relevant features for improving model accuracy.

Python Development
  • Utilize Python programming language, with a focus on PyTorch or TensorFlow, for the development of machine learning solutions.

  • Develop and maintain efficient and scalable code for production deployment in the LangChain environment.

Tooling and Infrastructure
  • Work with cloud platforms (e.g. Azure) to deploy and manage machine learning models.

  • Implement and manage MLOps practices, including ML pipelines and CI/CD, for automated model deployment and monitoring.

Vector Database
  • Leverage vector databases to efficiently store and retrieve embeddings generated by ML models.

  • Optimize data storage and retrieval processes for large-scale ML model applications.

Collaboration and Communication
  • Collaborate with cross-functional teams, including data scientists, software engineers, and product managers, to understand project requirements and deliver high-quality ML solutions.

  • Communicate complex technical concepts related to ML model development to both technical and non-technical stakeholders.

Research and Innovation
  • Stay up-to-date with the latest advancements in ML models, and related fields.

  • Contribute to the development and implementation of new techniques and technologies to improve language model performance.

Documentation
  • Maintain thorough documentation of code, algorithms, and methodologies specific to ML model development.

  • Provide documentation and training to other team members as needed.

Mentorship and Leadership
  • Provide mentorship and guidance to junior members of the team in the area of ML model development.

  • Take a leadership role in shaping the technical direction of machine learning projects within the NLP domain.

Qualifications:

  • Master’s or Ph.D. in Computer Science, Machine Learning, or a related field.

  • At least 6 years of experience in developing and deploying ML models. Strong proficiency in Python, especially with Tensorflow/PyTorch, and familiarity with relevant libraries and frameworks.

  • Experience with LangChain, vector databases, and MLOps practices. Proven track record of successfully delivering ML model solutions in a production environment.

  • Excellent problem-solving and critical-thinking skills. Strong communication and collaboration skills.