CrawlJobs Logo

Senior ML Engineer- Distillation

amd.com Logo

AMD

Location Icon

Location:
Poland , Warsaw

Category Icon
Category:
IT - Software Development

Job Type Icon

Contract Type:
Not provided

Salary Icon

Salary:

Not provided
Save Job
Save Icon
Apply Position

Job Description:

A seasoned Senior ML Engineer who drives distillation of ML Models for high-performance, production-ready rendering systems. You are passionate about software engineering and possess leadership skills to drive sophisticated issues to resolution. Able to communicate effectively and work optimally with different teams across AMD.

Job Responsibility:

  • Distillation and compression: KD variants, hint/fitnets, attention transfer, feature mimicking, low-rank/SVD, sparsity
  • Efficient architectures: MobileNet/EfficientNet, vision transformers optimization, lightweight diffusion/UNet variants, NeRF/instant-NGP distillation
  • Inference optimization: TensorRT, CUDA, cuDNN, ONNX, quantization-aware training, weight clustering, operator fusion
  • Metrics: SSIM, LPIPS, PSNR, FID/KID, latency/throughput profiling, memory/activation footprint analysis
  • Data and training: large-scale dataset curation, synthetic data generation, curriculum learning, augmentation strategies
  • MLOps: experiment tracking, CI/CD for models, model registries, reproducibility, telemetry
  • Integrate ML inference into production rendering pipelines: define model I/O, preprocessing/postprocessing, and make trade-offs for latency, throughput, and quality
  • Collaborate across teams (ML researchers, engine/platform, tooling, QA) to translate ML and product requirements into graphics-friendly implementations and integration plans
  • Mentor other engineers, conduct code reviews, and help define best practices for rendering, performance, and SDK delivery

Requirements:

  • 6–10+ years in ML engineering or applied research
  • 3+ years focused on model distillation/compression at production scale
  • Strong proficiency in PyTorch (preferred) or JAX/TF
  • Ability to implement custom training loops, distributed training, and mixed precision
  • Demonstrated experience shipping distilled or compressed models to production with measurable gains in latency/memory and maintained quality
  • Deep understanding of knowledge distillation techniques: teacher–student frameworks, soft-labels, intermediate feature matching, contrastive distillation, task-specific loss shaping
  • Hands-on experience with quantization (static/dynamic, PTQ/QAT), pruning, and graph-level optimizations (operator fusion)
  • GPU performance engineering: CUDA fundamentals, TensorRT/ONNX Runtime, kernel profiling (Nsight), memory/layout optimization
  • Solid grasp of computer graphics fundamentals: rendering pipeline, shaders, sampling, anti-aliasing, tone mapping, and perceptual metrics
  • Strong software engineering: Python/C++ proficiency, testing, code quality, version control, reproducible pipelines, containerization
  • Cross-functional leadership and communication
  • ability to drive roadmaps and align stakeholders across ML, graphics, and product
  • Bachelor’s or Master's degree in Computer Science, Mathematics, or equivalent

Additional Information:

Job Posted:
April 05, 2026

Work Type:
Remote work
Icon
AMD - All Job Offers
Job Link Share:
Copy Link Icon
Facebook Icon Twitter Icon LinkedIn Icon

Looking for more opportunities? Search for other job offers that match your skills and interests.

Briefcase Icon

Similar Jobs for Senior ML Engineer- Distillation

Senior Software Engineer – AI

NStarX is seeking a highly skilled Senior Software Engineer – AI with a strong f...
Location
Location
India , Hyderabad
Salary
Salary:
Not provided
nstarxinc.com Logo
NStarX
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor’s or Master’s degree in Computer Science, Machine Learning, Data Science, or a related field (PhD is a plus)
  • 9+ years of experience in AI/ML engineering or related roles
  • 3+ years of experience in Generative AI with team leadership responsibilities
  • Proven track record of production-grade ML and GenAI model development and deployment
  • Programming: Python (preferred)
  • GenAI Frameworks: Hugging Face Transformers, Diffusers, LangChain, TGI
  • Serving & Inference: FastAPI, gRPC, NVIDIA Triton, TorchServe
  • Cloud Platforms: AWS (SageMaker, EKS), GCP (Vertex AI, GKE), Azure (Azure ML, AKS)
  • MLOps & DevOps: Kubeflow, MLflow, GitHub Actions, Jenkins, Helm, Terraform
  • Optimization Techniques: Model quantization, distillation, pipeline and tensor parallelism
Job Responsibility
Job Responsibility
  • Design, develop, and deploy machine learning models and AI algorithms to address complex business challenges
  • Lead and mentor a team of AI/ML engineers, ensuring quality and scalability in solution design and implementation
  • Collaborate closely with cross-functional teams including data scientists, software engineers, product managers, and UX designers
  • Lead the development and deployment of Generative AI applications across text, code, image, and audio modalities using state-of-the-art LLMs
  • Design and implement CI/CD pipelines for the GenAI model lifecycle including training, validation, packaging, and deployment
  • Apply best practices for model performance tuning, cost optimization, and scalable deployment in cloud and hybrid environments
  • Develop prompt engineering, fine-tuning strategies (LoRA, QLoRA, PEFT), and evaluation protocols tailored to business use cases
  • Stay current with emerging trends in AI, ML, and Generative AI and drive adoption across teams
  • Document processes, model architectures, and deployment strategies for traceability and knowledge sharing
  • Work closely with cross-functional teams to gather requirements and deliver high-quality solutions
What we offer
What we offer
  • Competitive salary aligned with market standards
  • Opportunities for professional development and skill enhancement
  • A collaborative and innovative work environment
  • Fulltime
Read More
Arrow Right

Senior Software Engineer | Computer Vision & Machine Learning

We are looking for a seasoned, hands-on Software Engineer with deep expertise in...
Location
Location
Germany; Spain , Mannheim; Valencia
Salary
Salary:
Not provided
hubblr.io Logo
HUBBLR GmbH
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 4+ years of professional experience building ML/CV applications in Python
  • Proven expertise in Computer Vision: Demonstrable experience with object detection, image segmentation, or related tasks in a production environment
  • Deep Learning Proficiency: Hands-on experience with modern frameworks like PyTorch or TensorFlow
  • CV Toolkit: Strong skills with libraries such as OpenCV, scikit-image, or similar
  • Technical Leadership: Experience leading at least one major ML/CV project from initial concept to successful production deployment
  • Model Optimization: Practical knowledge of techniques like quantization, pruning, or knowledge distillation to make models efficient
  • Pragmatic Mindset: An analytical, MVP-focused approach to problem-solving, with a focus on delivering value quickly
  • Excellent Communication: Fluent in English, with the ability to provide constructive feedback to teammates and communicate clearly with clients
  • Location: Ability to work from our Mannheim or Valencia office in a hybrid model
Job Responsibility
Job Responsibility
  • Lead Technical Development: Design, build, and deploy production-ready computer vision solutions for real-world applications like object detection, image segmentation, and theft detection
  • Own the ML Lifecycle: Take full ownership of the end-to-end ML pipeline, including data annotation workflows (e.g., CVAT), model training, optimization (quantization, pruning), and deployment on cloud infrastructure
  • Drive Architectural Decisions: Actively participate in and influence architectural and technological choices for new and existing ML-powered software projects
  • Collaborate and Guide: Work within a distributed team (Münster, Hamburg, Mannheim, Valencia), provide technical guidance, conduct code reviews, and champion best practices
  • Act as a Client Partner: Communicate directly with client CTOs and stakeholders. You'll provide status updates, estimate development tasks, and help manage project scope and timelines
What we offer
What we offer
  • High Impact: You won't be a small cog in a big machine. You will own projects, see your work go live, and make a tangible impact on our clients' businesses
  • Growth & Learning: You'll be exposed to a variety of projects across different industries, ensuring you're always learning and tackling new challenges
  • A Culture of Trust & Ownership: We hire smart people and trust them. You'll have the autonomy to make decisions and manage your work independently
  • Diverse & Inclusive Team: We are committed to building a diverse team. We welcome people who are like us and people who are unlike us. This includes but is not limited to gender, ethnicity, age, academic background, and character
  • Fulltime
Read More
Arrow Right

Senior Principal Machine Learning Engineer - LLM Post-Training and Optimization

Atlassian is seeking a highly skilled and experienced Senior Principle Machine L...
Location
Location
United States , Mountain View
Salary
Salary:
243100.00 - 407200.00 USD / Year
https://www.atlassian.com Logo
Atlassian
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Ph.D. or Master’s degree in Computer Science, Machine Learning, Artificial Intelligence, or a related field
  • 8+ years of experience in machine learning, with a focus on large-scale model development and optimization
  • Deep expertise in LLM and transformer architectures (e.g., GPT, BERT, T5)
  • Strong proficiency in Python and ML frameworks such as PyTorch, JAX, or TensorFlow
  • Experience with distributed training techniques and large-scale data processing pipelines
  • Proven track record of deploying machine learning models in production environments
  • Familiarity with model optimization techniques, including quantization, pruning, and knowledge distillation
  • Strong problem-solving skills and ability to work in a fast-paced, collaborative environment
  • Excellent communication skills and ability to translate technical concepts for diverse audiences
Job Responsibility
Job Responsibility
  • Lead the fine-tuning and post-training optimization of large language models (LLMs) for diverse applications
  • Develop and implement techniques for model compression, quantization, pruning, and knowledge distillation to optimize performance and reduce computational costs
  • Conduct research on advanced techniques in transfer learning, reinforcement learning, and prompt engineering for LLMs
  • Design and execute rigorous benchmarking and evaluation frameworks to assess model performance across multiple dimensions
  • Collaborate with infrastructure teams to optimize LLM deployment pipelines, ensuring scalability and efficiency in production environments
  • Stay at the forefront of advancements in LLM technologies, sharing insights, driving innovation within the team, and leading agile development
  • Mentoring other team members, facilitating within/across team workshops, fostering a culture of technical excellence and continuous learning
What we offer
What we offer
  • health coverage
  • paid volunteer days
  • wellness resources
  • Fulltime
Read More
Arrow Right

Senior ML Engineer

We are looking for a highly motivated GenAI Engineer to join the Customer Obsess...
Location
Location
India , Bangalore
Salary
Salary:
Not provided
uber.com Logo
Uber
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's or Master's in Computer Science, Statistics, or a related field or Equivalent Experience
  • Minimum 5 years of experience in industry with a strong focus on machine learning and optimization
  • Experience with ML packages such as Tensorflow, PyTorch, JAX, and Scikit-Learn
  • Solid understanding of statistical analysis and feature engineering techniques
  • Excellent communication and collaboration skills
  • Ability to work independently and take ownership of projects
  • Experience using SQL in a production environment
  • Experience in experimental design and analysis, exploratory data analysis, and statistical analysis
  • Experience with dashboarding and using data visualization tools
  • Experience using statistical methodologies such as sampling, statistical estimates, descriptive statistics, or similar
Job Responsibility
Job Responsibility
  • Design, develop, and productionize machine learning (ML) solutions in the field of customer support engineering spanning generative AI algorithms, agentic AI design at scale, NLP for query understanding and ranking responses, distillation techniques, etc.
  • Productionize and deploy these models for real-world applications in customer support
  • Design and analyze experiments using a combination of data analysis/statistical analysis to lead the team to a reasonable inference
  • Review code and designs of teammates, providing constructive feedback
  • Collaborate with cross-functional teams to brainstorm new solutions and iterate on the product
  • Mentor junior engineers
  • Fulltime
Read More
Arrow Right

Senior ML Engineer

We are looking for a highly motivated GenAI Engineer to join the Customer Obsess...
Location
Location
India , Bangalore
Salary
Salary:
Not provided
uber.com Logo
Uber
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's or Master's in Computer Science, Statistics, or a related field or Equivalent Experience in Conversational GenAI
  • Minimum 8 years of experience in industry with a strong focus on machine learning and optimization
  • Experience with ML packages such as Tensorflow, PyTorch, JAX, and Scikit-Learn
  • Solid understanding of statistical analysis and feature engineering techniques
  • Excellent communication and collaboration skills
  • Ability to work independently and take ownership of projects
  • Experience using SQL in a production environment
  • Experience in experimental design and analysis, exploratory data analysis, and statistical analysis
  • Experience with dashboarding and using data visualization tools
  • Experience using statistical methodologies such as sampling, statistical estimates, descriptive statistics, or similar
Job Responsibility
Job Responsibility
  • Design, develop, and productionize machine learning (ML) solutions in the field of customer support engineering spanning generative AI algorithms, agentic AI design at scale, NLP for query understanding and ranking responses, distillation techniques, etc.
  • Productionize and deploy these models for real-world applications in customer support
  • Design and analyze experiments using a combination of data analysis/statistical analysis to lead the team to a reasonable inference
  • Review code and designs of teammates, providing constructive feedback
  • Collaborate with cross-functional teams to brainstorm new solutions and iterate on the product
  • Mentor junior engineers
Read More
Arrow Right

Senior Applied Machine Learning Engineer - Asset Intelligence

We are seeking a highly skilled and motivated Senior Applied Machine Learning En...
Location
Location
United States , San Francisco
Salary
Salary:
Not provided
getmaintainx.com Logo
MaintainX
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 7+ years of experience in Machine Learning, Data Science, or Applied AI
  • Expertise in Python, and strong familiarity with PyTorch, TensorFlow, and cloud ML stacks (AWS, Databricks, or similar)
  • Proven experience deploying production ML systems at scale
  • Strong background in LLMs, time-series modeling, and anomaly detection for real-world data
  • Demonstrated ability to lead architectural decisions, mentor engineers, and collaborate across product, data, and platform teams
  • Knowledge of MLOps tooling (Docker, Kubernetes, Weights & Biases, MLflow, SageMaker)
  • Advanced degree (MS/PhD) in Computer Science, Machine Learning, or related field preferred
Job Responsibility
Job Responsibility
  • Lead technical direction for predictive maintenance, anomaly detection, and LLM-powered intelligence across MaintainX products
  • Architect end-to-end ML systems—from data ingestion and feature engineering to model training, deployment, and monitoring
  • Mentor a growing team of ML and data engineers, instilling best practices for experimentation, evaluation, and model lifecycle management
  • Partner with product and engineering leaders to align AI roadmap with customer needs and business goals
  • Design reliable data and feedback loops that connect customer telemetry and operator feedback to model retraining
  • Drive performance optimization through techniques like quantization, distillation, and scalable inference serving
  • Work with LLM frameworks (LangChain, LlamaIndex, Hugging Face) to build reasoning systems and agentic workflows for asset and work intelligence
  • Ensure ML infrastructure meets production standards for latency, reliability, explainability, and security
What we offer
What we offer
  • Competitive salary and meaningful equity opportunities
  • Healthcare, dental, and vision coverage
  • 401(k) / RRSP enrollment program
  • Take what you need PTO
  • A high impact Culture: You’ll work with Smart, Humble Optimists across the globe
  • Meritocratic environment where ideas and outcomes are publicly celebrated
  • Fulltime
Read More
Arrow Right

Senior Applied Machine Learning Engineer - Asset Intelligence

We are seeking a highly skilled and motivated Senior Applied Machine Learning En...
Location
Location
Canada , Montreal, Toronto, Vancouver
Salary
Salary:
Not provided
getmaintainx.com Logo
MaintainX
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 7+ years of experience in Machine Learning, Data Science, or Applied AI
  • Expertise in Python, and strong familiarity with PyTorch, TensorFlow, and cloud ML stacks (AWS, Databricks, or similar)
  • Proven experience deploying production ML systems at scale
  • Strong background in LLMs, time-series modeling, and anomaly detection for real-world data
  • Demonstrated ability to lead architectural decisions, mentor engineers, and collaborate across product, data, and platform teams
  • Knowledge of MLOps tooling (Docker, Kubernetes, Weights & Biases, MLflow, SageMaker)
  • Advanced degree (MS/PhD) in Computer Science, Machine Learning, or related field preferred
Job Responsibility
Job Responsibility
  • Lead technical direction for predictive maintenance, anomaly detection, and LLM-powered intelligence across MaintainX products
  • Architect end-to-end ML systems—from data ingestion and feature engineering to model training, deployment, and monitoring
  • Mentor a growing team of ML and data engineers, instilling best practices for experimentation, evaluation, and model lifecycle management
  • Partner with product and engineering leaders to align AI roadmap with customer needs and business goals
  • Design reliable data and feedback loops that connect customer telemetry and operator feedback to model retraining
  • Drive performance optimization through techniques like quantization, distillation, and scalable inference serving
  • Work with LLM frameworks (LangChain, LlamaIndex, Hugging Face) to build reasoning systems and agentic workflows for asset and work intelligence
  • Ensure ML infrastructure meets production standards for latency, reliability, explainability, and security
What we offer
What we offer
  • Competitive salary and meaningful equity opportunities
  • Healthcare, dental, and vision coverage
  • 401(k) / RRSP enrollment program
  • Take what you need PTO
Read More
Arrow Right

Senior Software Developer

We’re looking for a Senior Software Developer to be part of our success story. W...
Location
Location
United Kingdom
Salary
Salary:
Not provided
activate-group.com Logo
Activate Group Limited
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Expert level Python with 5+ years of production experience
  • 3+ years experience with PyTorch and/or TensorFlow
  • Strong proficiency with scikit-learn, XGBoost, LightGBM
  • Expert level data processing with pandas, NumPy, Spark
  • Practical experience with CNNs, RNNs, Transformers
  • Git proficiency with focus on ML workflows
  • Experience with LLMs (GPT, Claude, Llama) and prompt engineering
  • Experience with vector databases (Pinecone, Weaviate, Chroma, or similar)
  • RAG (Retrieval Augmented Generation) implementation experience
  • Experience with Computer Vision libraries (OpenCV, PIL, torchvision)
Job Responsibility
Job Responsibility
  • Design, develop, and deploy machine learning models and pipelines using Python
  • Build and maintain end-to-end ML systems from data ingestion to model serving
  • Implement deep learning solutions using PyTorch and TensorFlow
  • Develop and optimize NLP solutions and computer vision applications
  • Create scalable feature engineering and data preprocessing pipelines
  • Build model training, evaluation, and monitoring frameworks
  • Implement MLOps practices for continuous model improvement
  • Design and maintain vector databases for similarity search and RAG applications
  • Integrate LLMs and foundation models into production applications
  • Optimize model inference for latency and throughput requirements
What we offer
What we offer
  • 33 days holiday (including bank holidays)
  • Personal health cash plan – claim back the cost of things like dentist and optical check ups
  • Enhanced maternity / paternity / adoption / shared parental pay
  • Life assurance: three times basic salary
  • Free breakfasts and fruit
  • Birthday surprise for everybody
  • Fulltime
Read More
Arrow Right
Remote jobs Jobs by Category
Art and Entertainment Jobs Banking Jobs Call Center Jobs Construction Jobs Consulting Jobs Customer Service Jobs Education Jobs Energy Jobs Finance Jobs Gastronomy Jobs Health and Beauty Jobs Hospitality and Tourism Jobs Human Resources Jobs Insurance Jobs IT - Administration Jobs IT - Software Development Jobs Legal Jobs Logistics Jobs Manufacturing Jobs Marketing, Digital Marketing, SEO, SEM Jobs Media Jobs Nursing Jobs Office Administration Jobs Pharmacy Jobs Physical Work Jobs Public Relations Jobs Public Sector Jobs Purchasing Jobs Quality Control Jobs Real Estate Jobs Research and Development Jobs Sales Jobs
Job Positions Jobs by Country
United States Jobs United Kingdom Jobs India Jobs Canada Jobs Australia Jobs Germany Jobs Ireland Jobs Poland Jobs
Company
Blog Salaries About Us Contact
Follow Us
Facebook
X (Twitter)
LinkedIn
Instagram
CrawlJobs Group
ITFlashcards AllDevBlogs
© 2026 CrawlJobs Ltd. All Rights Reserved
Terms & Conditions Privacy Policy