General AI Specialist - PyTorch, MLflow, & Transformers at OneSeven Tech

OneSeven Tech; United States; Remote; USD 140,000 - 210,000; Full time

Benefits offered by this job

Remote work

Contract-to-full-time

Paid time off

Referral bonus

Job summary

OneSeven Tech is seeking an experienced machine learning engineer to design and productionize deep learning and Transformer models. You will build robust ML systems, profile performance, and ensure secure API serving in a remote, international setting.

The role emphasizes reproducible experiment workflows with MLflow/DVC, collaboration in fast delivery cycles, and hands-on work with PyTorch or TensorFlow and the Hugging Face ecosystem. Remote work from anywhere in Latin America is supported.

Qualifications

• Proficiency with PyTorch or TensorFlow and Hugging Face.
• Advanced Python including OO and asynchronous programming.
• Expert knowledge of MLflow Tracking, Projects, and Models.
• Experience with Git workflows and DVC.
• Strong grounding in attention mechanisms and linear algebra.
• Docker proficiency and building secure model-serving APIs.

Responsibilities

• Design and implement deep learning and Transformer models, comparing results with classical ML baselines.
• Create reproducible experiment and model lifecycle workflows using MLflow and DVC.
• Build, profile, and secure REST or gRPC APIs for production model serving.
• Collaborate across disciplines in short, one-week delivery cycles to bring robust systems into use.

Skills

PyTorch

TensorFlow

Hugging Face

Python advanced

Async programming

MLflow

Git workflows

DVC

Docker

Attention mechanisms

Linear algebra

Tools

MLflow

DVC

REST APIs/gRPC

Job description

Role overview

Develop and productionize machine learning systems, with a focus on deep learning and Transformer models. The work combines model development and mathematical foundations with reproducible MLOps workflows and secure, performant model serving, moving solutions from research into practical delivery.

Responsibilities

• Design and implement deep learning and Transformer models, comparing results with classical machine learning baselines.
• Create reproducible experiment and model lifecycle workflows using MLflow and DVC.
• Build, profile, and secure REST or gRPC APIs for production model serving.
• Collaborate across disciplines in short, one-week delivery cycles to bring robust systems into use.

Requirements

• Strong experience with PyTorch or TensorFlow/Keras and the Hugging Face ecosystem, including Transformers, Datasets, or Accelerate.
• Advanced Python skills, including object-oriented and asynchronous programming.
• Expert knowledge of MLflow Tracking, Projects, and Models.
• Experience with advanced Git workflows and DVC.
• Strong grounding in attention mechanisms, gradient descent, and linear algebra.
• Proficiency with Docker and building secure model-serving APIs.

Nice to have

• Familiarity with retrieval-augmented generation and vector databases.
• Experience with MLflow Recipes, Kubernetes, distributed training, or model optimization tools such as DeepSpeed, ONNX Runtime, or TensorRT.
• Hands-on experience with a managed cloud machine learning platform.

Benefits and work setup

• Fully remote work from anywhere in Latin America.
• Long-term contract beginning with a three-month period, with an expectation of moving to full-time afterward.
• Paid time off and a referral bonus program.