NLP Machine Learning Engineer

Machine Learning Engineers play a critical role at Columb Labs, they are responsible for implementing the core logic for our clients' projects. They are known both for their excellent software engineering skills and solid knowledge of deep learning and machine learning algorithms.

We are constantly striving to boost our clients' business KPIs. You will implement state-of-the-art machine learning solutions and bring them to businesses. Responsibilities include researching the latest machine learning papers, implementation and optimization of machine learning models, integration of your solutions and communication with clients.

  • 4+ years working with production machine learning
  • ML/DL Stack: pytorch, sklearn
  • Experience with transformer-based architectures: GPT-2, BERT
  • Experience with model finetuning
  • Excellent Python knowledge for writing scalable and effective production code
  • Excellent software engineering skills: CI/CD, Docker, pytest, code documentation
  • Good knowledge of computer science, data structures, and algorithms
  • Exceptional problem solving and ability to work independently
  • Fluency in both oral and written English
Nice to have:
  • Experience optimizing deep learning models
  • Experience with pytorch-lightning, transformers, ONNX, Triton
  • Machine learning or deep learning publications
  • Experience building scalable systems with cloud infrastructures like AWS
  • Contribution to NLP open-source projects
Why you should work here:
  • Work with a strong team (ex Yandex, CERN, Columbia University)
  • No legacy, contemporary tech stack
  • Competitive salaries
  • Remote work