General AI Specialist - PyTorch, MLflow, & Transformers

OneSeven Tech (OST); Buenos Aires; A distancia; ARS 133.982.000 - 178.643.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Fully Remote

Paid PTO

Referral Program

Referral Program

Descripción de la vacante

A leading digital product agency based in Latin America is seeking a high-caliber AI Specialist to develop advanced machine learning systems. This fully remote role focuses on building state-of-the-art models using Deep Learning and ensuring reproducibility through expert MLOps. The candidate will work on cutting-edge technology, engage in rigorous mathematical applications, and contribute to robust AI solutions. Strong proficiency in PyTorch and expert knowledge of MLflow are essential. Competitive compensation and long-term contracts offered.

Formación

• Strong proficiency in PyTorch or TensorFlow/Keras and experience with Hugging Face ecosystem.
• Expert knowledge of MLflow for lifecycle management.
• Expert-level Python and advanced Git workflows.
• Deep theoretical understanding of the Attention Mechanism, Gradient Descent, and Linear Algebra.

Responsabilidades

• Design and implement Deep Learning models and Transformers.
• Lead implementation of reproducible ML environments using MLflow and DVC.
• Build, profile, and secure high-performance APIs.
• Cross-Functional Delivery: Work within 1-week sprints to move fast from research to delivery, ensuring robust AI solutions.

Conocimientos

Strong proficiency in PyTorch

Hands-on experience with TensorFlow/Keras

Expert knowledge of MLflow

Expert-level Python

Deep theoretical understanding of Attention Mechanism

Proficiency with Docker

Python (OOP, async)

Herramientas

Docker

AWS SageMaker

GCP Vertex AI

Azure ML

gRPC

Kubernetes

Descripción del empleo

Project Brief

We are seeking a high-caliber AI Specialist to build and scale advanced machine learning systems. You will be responsible for developing state-of-the-art models using Deep Learning and Transformer architectures while ensuring absolute reproducibility through expert-level MLOps. This role is at the intersection of rigorous mathematics and elite engineering, focusing on creating elegant solutions to complex AI problems.

Project Brief

We are seeking a high-caliber AI Specialist to build and scale advanced machine learning systems. You will be responsible for developing state-of-the-art models using Deep Learning and Transformer architectures while ensuring absolute reproducibility through expert-level MLOps. This role is at the intersection of rigorous mathematics and elite engineering, focusing on creating elegant solutions to complex AI problems.

Location: Remote LATAM

Must-Haves

Machine Learning & Frameworks: Strong proficiency in PyTorch or TensorFlow/Keras and hands-on experience with the Hugging Face ecosystem (Transformers, Datasets, Accelerate).

MLflow Mastery: Expert knowledge of MLflow Tracking, Projects, and Models for full lifecycle management.

Engineering Excellence: Expert-level Python (OOP, asynchronous programming) and advanced Git workflows including DVC.

Core Mathematics: Deep theoretical understanding of the Attention Mechanism, Gradient Descent, and Linear Algebra.

Deployment & Tooling: Proficiency with Docker and building/securing REST/gRPC APIs for model serving.

Nice to Have

Generative AI: Familiarity with RAG architectures and vector databases such as Pinecone, Milvus, or Weaviate.

Advanced MLOps: Experience with MLflow Recipes for standardized production workflows and basic Kubernetes orchestration.

Efficiency & Scale: Experience with DeepSpeed, ONNX Runtime, or NVIDIA TensorRT, and knowledge of distributed training (Ray, Horovod, or DDP).

Cloud Platforms: Hands-on experience with AWS SageMaker, GCP Vertex AI, or Azure ML.

What You Will Do

Model Development: Design and implement Deep Learning models and Transformers, using classical ML (Scikit-learn, XGBoost) for baseline comparisons.

Standardize Workflows: Lead the implementation of reproducible ML environments using MLflow and Data Version Control (DVC).

Production Engineering: Build, profile, and secure high-performance APIs to serve models in production environments.

Cross-Functional Delivery: Work within 1-week sprints to move fast from research to delivery, ensuring robust and scalable AI solutions.

Why This Could Be Your Next Big Move

Work on Cutting-Edge Tech at Scale - This is a unique opportunity to join an elite digital product agency where you will apply deep mathematical concepts to real-world products. You will have high influence over the AI architecture and MLOps standards, working in a fast-paced environment that prioritizes technical excellence and elegant engineering. If you are looking to bridge the gap between research-level AI and production-grade delivery, this is the place to do it.

Benefits & Compensation

Fully Remote - work from anywhere in Latin America.

Long-term contract - starting with a 6-month contract, then full-time.

Paid PTO - details provided per role.

Referral Program - earn a bonus for referring talent that gets hired.

To Apply

Please send your resume in English.

LinkedIn Profile URL (Required).

GitHub Repository - mandatory for this engineering-heavy AI role (Required).

Cover Letter (Optional but encouraged) - tell us about the most complex Attention-based model you have deployed to production.

About OneSeven Tech

Founded by James Sullivan, OneSeven Tech is a premier digital product agency serving startups and enterprises. Our clients have collectively raised over $100M in VC, and our enterprise partners include NASDAQ-listed companies. We work fully remote, move fast with 1-week sprints, and focus on elegant solutions to complex problems.