Senior Data Scientist with PyTorch

We are seeking a Senior Data Scientist with PyTorch expertise to join a team of Geoscientists, Geophysicists, Data Scientists and Machine Learning Engineers, working to understand technical and business needs and translate them into reliable, high-impact deep learning solutions for seismic data. This role will also focus on improving existing seismic ML solutions and identifying new business use cases.Experience the freedom of remote work from anywhere in Georgia, whether from the comfort of your home, our modern offices in Tbilisi and Batumi or a coworking space in Kutaisi.ResponsibilitiesApply and develop deep learning models (U-Nets, CNNs and related architectures) for denoising and structural interpretation of 2D and 3D seismic dataImprove the accuracy, robustness and performance of existing seismic ML solutions, including model retraining, architecture refinement and hyperparameter tuningDevelop models for structural interpretation tasks such as fault detection, horizon tracking and seismic facies classificationWork closely with Geoscientists and Geophysicists to validate model outputs against expert interpretation and well data, and incorporate their feedback into model improvementsIdentify, evaluate and propose new business use cases where computer vision and deep learning can add value across exploration and production workflowsStay up to date with research in computer vision and seismic ML, and assess new methods for practical applicationFollow good engineering practices for reproducible experiments, model versioning and deployment of models into production workflowsRequirementsPreferably 2-4 years of experience in similar roles (ML Engineer, Data Scientist or Geophysicist with a strong ML focus), including hands-on experience applying deep learning to real-world image or volumetric dataUnderstanding of Computer Vision, including image segmentation, denoising, feature extraction and convolutional architecturesProficiency in PyTorch for building, training and evaluating deep learning modelsExpertise in U-Net, CNN and related architectures for 2D and 3D seismic dataBackground in the Upstream Oil & Gas domain, working with Geoscience and Seismic data and teamsUnderstanding of seismic data concepts such as acquisition and processing basics, seismic attributes, noise types and structural features (faults, horizons)Strong Python skillsEnglish proficiency at B2 level or higherNice to haveExperience with transformer-based or self-supervised models for vision or seismic dataFamiliarity with MLOps tools (e.g. MLflow), Docker and cloud AI services (Azure, AWS or GCP)We offerWe connect like-minded peopleDelivering innovative solutions to industry leaders, making a global impactEnjoyable working environment, whether it is the vibrant office or the comfort of your own homeOpportunity to work abroad for up to two months per yearRelocation opportunities within our offices in 55+ countriesCorporate and social eventsWe invest in your growthLeadership development, career advising, soft skills and well-being programsCertifications, including GCP, Azure and AWSUnlimited access to EPAM's internal learning databaseFree English classes with certified teachersWe cover it allParticipation in the Employee Stock Purchase PlanMonetary bonuses for engaging in the referral programComprehensive medical & family care packageFive trust days per year (sick leave without a medical certificate)Benefits package (sports activities, a variety of stores and services)EPAM is global leader in AI transformation engineering and integrated consulting, serving Forbes Global 2000 companies and ambitious startups. With over thirty years of expertise in custom software, product and platform engineering, we empower our clients to become AI-Native enterprises, driving measurable value from innovation and digital investments.