CrawlJobs Logo

AI Framework Engineer

amd.com Logo

AMD

Location Icon

Location:
China , Shanghai

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:

As a core member of the team, you will play a pivotal role in optimizing and developing deep learning frameworks for AMD GPUs. Your experience will be critical in enhancing GPU kernels, deep learning models, and training/inference performance across multi-GPU and multi-node systems. You will engage with both internal GPU library teams and open-source maintainers to ensure seamless integration of optimizations, utilizing cutting-edge compiler technologies and advanced engineering principles to drive continuous improvement.

Job Responsibility:

  • Build and optimize end to end distributed inference (e.g, P/D disaggregation and Large-EP) and RL solutions on mainstream frameworks like vLLM and SGlang
  • Collaborate with internal GPU library teams to analyze and improve training and inference performance on AMD GPUs
  • Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream
  • Optimize deep learning performance on both scale-up (multi-GPU) and scale-out (multi-node) systems
  • Leverage advanced compiler technologies to improve deep learning performance
  • Enhance the full pipeline, including integrating graph compilers
  • Apply sound engineering principles to ensure robust, maintainable solutions

Requirements:

  • Bachelor's and/or Master's in Computer Science, Computer Engineering, Electrical Engineering, or related fields
  • 3+ years of professional experience in technical software development, with a focus on GPU optimization, performance engineering, and framework development

Nice to have:

Text to Video or Image to Video experience

Additional Information:

Job Posted:
May 13, 2026

Employment Type:
Fulltime
Work Type:
On-site 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 AI Framework Engineer

Digital Twin AI Framework Engineer

This is a career-defining opportunity to play a crucial role in a hyper-scale AI...
Location
Location
United States , Salt Lake City
Salary
Salary:
Not provided
passivelogic.com Logo
PassiveLogic
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • MS/PhD in CS/CE/EE/Mathematics, information science, computer science, computational linguistics, computational physiology, human comfort, physics, or equivalent experience
  • Demonstrated experience in developing mappings and transformation processes to take data from a relational paradigm into a knowledge graph environment
  • Enthusiasm for human-centered API design and object model design
  • Demonstrated expertise in multi-disciplinary system modeling
  • Strong theoretical background in systems theory and graph theory
  • Strong programming and software architecture skills (C++, Swift, Rust, etc)
  • Exceptional communication skills
  • Organized and strategic
  • Collaborative mindset
  • Adaptability
Job Responsibility
Job Responsibility
  • Develop and evolve the Quantum digital twin object model, ontology, and API
  • Develop the Quantum Knowability Model — evolve the formal proofs of digital twin definitions, systems, events, and compositional stage by which the formalisms can be known
  • Support the development of the digital twin AI inferencing framework and the accelerator compute team
  • Advocate for human-centered UX/UI at every stage of ontology development
  • Apply systems theory and design processes throughout the Quantum evolution process, always collecting feedback and improving the language
  • Collaborate and coordinate with Quantum working groups including: formal methods, AI, compute, physics, geometry, compiler, and semantic teams, as well as with other Quantum partners to address their feedback and incorporate their needs
What we offer
What we offer
  • Competitive compensation
  • Generous equity share package
  • Medical, dental and vision coverage
  • Disability and life Insurance options
  • Flex PTO
  • Team-building events
  • Free catered lunch in the office Monday — Friday
  • Free ski pass
  • Free National Park pass
  • Fulltime
Read More
Arrow Right

Senior Data Engineer – Data Engineering & AI Platforms

We are looking for a highly skilled Senior Data Engineer (L2) who can design, bu...
Location
Location
India , Chennai, Madurai, Coimbatore
Salary
Salary:
Not provided
optisolbusiness.com Logo
OptiSol Business Solutions
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Strong hands-on expertise in cloud ecosystems (Azure / AWS / GCP)
  • Excellent Python programming skills with data engineering libraries and frameworks
  • Advanced SQL capabilities including window functions, CTEs, and performance tuning
  • Solid understanding of distributed processing using Spark/PySpark
  • Experience designing and implementing scalable ETL/ELT workflows
  • Good understanding of data modeling concepts (dimensional, star, snowflake)
  • Familiarity with GenAI/LLM-based integration for data workflows
  • Experience working with Git, CI/CD, and Agile delivery frameworks
  • Strong communication skills for interacting with clients, stakeholders, and internal teams
Job Responsibility
Job Responsibility
  • Design, build, and maintain scalable ETL/ELT pipelines across cloud and big data platforms
  • Contribute to architectural discussions by translating business needs into data solutions spanning ingestion, transformation, and consumption layers
  • Work closely with solutioning and pre-sales teams for technical evaluations and client-facing discussions
  • Lead squads of L0/L1 engineers—ensuring delivery quality, mentoring, and guiding career growth
  • Develop cloud-native data engineering solutions using Python, SQL, PySpark, and modern data frameworks
  • Ensure data reliability, performance, and maintainability across the pipeline lifecycle—from development to deployment
  • Support long-term ODC/T&M projects by demonstrating expertise during technical discussions and interviews
  • Integrate emerging GenAI tools where applicable to enhance data enrichment, automation, and transformations
What we offer
What we offer
  • Opportunity to work at the intersection of Data Engineering, Cloud, and Generative AI
  • Hands-on exposure to modern data stacks and emerging AI technologies
  • Collaboration with experts across Data, AI/ML, and cloud practices
  • Access to structured learning, certifications, and leadership mentoring
  • Competitive compensation with fast-track career growth and visibility
  • Fulltime
Read More
Arrow Right

Applied AI Engineer - AI Solutions

As an Applied AI Engineer, you’ll research and utilize state-of-the-art Gen AI a...
Location
Location
United States , Redwood City; San Francisco
Salary
Salary:
172000.00 - 300000.00 USD / Year
snorkel.ai Logo
Snorkel AI
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • B.S. degree in a quantitative field such as Computer Science, Engineering, Mathematics, Statistics, or comparable degree/experience
  • 3+ years of customer-facing experience in the design and implementation of AI/ML solutions
  • Proficiency in Python, including strong grounding in software engineering fundamentals (e.g., modular design, testing, profiling, packaging) and experience with modern Python constructs and libraries for type validation and typed data modeling (e.g., pydantic), building type-safe systems (e.g., mypy), testing (e.g., pytest), packaging and environment configuration (e.g., poetry), API and service frameworks (e.g., FastAPI), serialization and structured data handling (e.g., msgspec), and orchestration tooling relevant to ML deployment (e.g., Ray, Airflow)
  • Expertise across the Applied AI stack, spanning classical ML libraries (e.g., scikit-learn), deep learning frameworks (e.g., PyTorch), foundation-model ecosystems (e.g., Hugging Face Transformers), vector/embedding tooling (e.g., FAISS), data processing frameworks (e.g., pandas, Spark), retrieval/RAG tooling (e.g., Chroma, Weaviate), synthetic dataset curation, evaluation workflows, and LLM orchestration, workflow, agent authoring tools (e.g., LlamaIndex, LangGraph, CrewAI)
  • Experience leading strategic, customer-facing initiatives and collaborating with business stakeholders to ensure ML solutions drive successful business outcomes, with a strong focus on teaching and enablement
  • Outstanding presentation skills to technical and executive audiences, whether impromptu on a whiteboard or using presentations and demos
  • Ability to work in a fast-paced environment and balance priorities across multiple projects at once
Job Responsibility
Job Responsibility
  • Partner with customers to build and deploy impactful Gen AI and machine learning solutions, from use case scoping and data exploration to model development and deployment. This may involve leveraging Snorkel Flow or designing custom approaches using state-of-the-art tools, with the goal of delivering real business value and informing the evolution of the Snorkel platform
  • Develop and implement state of the art AI systems such as retrieval-augmented generation (RAG), fine-tuning pipelines, prompt engineering recipes and agentic workflows
  • Create augmented real-world datasets and comprehensive evaluation workflows to ensure model reliability, transparency, and stakeholder trust. A data- and evaluation-first mindset is essential for success in this role
  • Forge and manage relationships with our customers’ leadership and stakeholders to ensure successful development and deployment of AI projects with Snorkel Flow
  • Collaborate closely with pre-sales Solutions and Product teams to map customer needs to existing capabilities, prioritize roadmap gaps, and guide successful project setup
  • Work with other Applied AI Engineers to standardize solutions and contribute to internal tooling and best practices
  • Lead stakeholder education on quantitative capabilities, helping them to understand the strengths and weaknesses of different approaches and what problems are best-suited for Snorkel AI
  • Serve as the voice of our customers for new AI paradigms, data science workflows, and share customer feedback to product teams
  • Conduct one-to-few and one-to-many enablement workshops to transfer knowledge to customers considering or already using Snorkel AI
  • Annual travel up to 25%
What we offer
What we offer
  • equity in the form of employee stock options
  • Fulltime
Read More
Arrow Right

Applied AI Engineer – Generative AI

Kodiak is redefining what AI can do in autonomy, and generative models play a cr...
Location
Location
United States , Mountain View
Salary
Salary:
150000.00 - 250000.00 USD / Year
kodiak.ai Logo
Kodiak Robotics
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • BS, MS, or PhD in AI, Computer Science, or a related field, or at least 2-3 years of industry experience
  • Experience with diffusion models, autoregressive learning, or generative AI methods
  • Proficiency in Python and deep learning frameworks like PyTorch or TensorFlow
  • Strong understanding of uncertainty modeling and self-supervised learning
  • Passion for building AI that continuously adapts and learns
Job Responsibility
Job Responsibility
  • Develop generative models that enhance adaptability and improve AI robustness
  • Research and implement diffusion-based, autoregressive, and variational techniques for real-world AI applications
  • Apply generative methods to anticipate challenges and reinforce decision-making
  • Build self-improving AI pipelines that continuously refine model understanding
  • Optimize models for efficient, scalable, and interpretable AI decision-making
What we offer
What we offer
  • Competitive compensation package including equity and biannual bonuses
  • Excellent Medical, Dental, and Vision plans through Kaiser Permanente, Anthem, and Guardian (including a medical plan with infertility benefits)
  • Flexible PTO and generous parental leave policies
  • Office perks: dog-friendly, free catered lunch, a fully stocked kitchen, and free EV charging
  • Long Term Disability, Short Term Disability, Life Insurance
  • Wellbeing Benefits - Headspace, One Medical, Gympass, Spring Health
  • Fidelity 401(k)
  • Commuter, FSA, Dependent Care FSA, HSA
  • Various incentive programs (referral bonuses, patent bonuses, etc.)
  • Fulltime
Read More
Arrow Right

Senior AI Engineer

We are seeking an innovative AI Engineer to join a brand new team focused on pro...
Location
Location
United Kingdom , London
Salary
Salary:
Not provided
https://www.citi.com/ Logo
Citi
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Proven experience as an AI Engineer with a significant delivery history
  • Strong expertise in multiple programming languages & frameworks
  • Experience and proven experience in using quantitative testing practice applied to the field of AI/ML for actionable Go/No-Go decisions of delivering software to production
  • Demonstrated expertise of developing on a range of architectures, ideally up to and including container-based micro-services with focus on scalability, reliability, maintainability, and high performance
  • Good understanding of SQL and NoSQL databases
  • Excellent communication and collaboration skills
  • A growth mindset and willingness to learn and adapt in a fast-paced environment
  • Passion about site reliability engineering and its impact on product development
  • Being connected to latest technologies, like Generative AI, and keen to put them in practice.
Job Responsibility
Job Responsibility
  • Understand the landscape, tooling and procedures used by developers at Citi and look for opportunities to reduce toil and aid simplification using Gen AI based solutions
  • Apply classic AI and novel Gen AI evaluation methodology to raise the quality and reliability bar for the software that you will deliver, as well to manage and mitigate risks that are specific/inherent to this field
  • Advice on Evaluation metrics, devise and implement Quantitative Testing Plans, and help evolve the existing approaches to AI evaluation
  • Work with a wide variety of Citi technology teams and help them drive towards everything-as-code and a codified controls environment
  • Collaborate with product and engineering teams to design, build and maintain scalable and reliable web applications and services
  • Be hands-on with coding and software design to ensure adherence to high quality standards and best practices
  • Mentor and nurture other engineers to help them grow their skills and expertise
  • Support and drive cultural change, including instigating critical thinking about controls and processes and encouraging a culture of continuous improvement.
What we offer
What we offer
  • 27 days annual leave (plus bank holidays)
  • A discretional annual performance-related bonus
  • Private Medical Care & Life Insurance
  • Employee Assistance Program
  • Pension Plan
  • Paid Parental Leave
  • Special discounts for employees, family, and friends
  • Access to an array of learning and development resources.
  • Fulltime
Read More
Arrow Right

AI Engineer

AI Engineer position at Inetum, a European leader in digital services, focusing ...
Location
Location
Romania , Bucharest
Salary
Salary:
Not provided
https://www.inetum.com Logo
Inetum
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Minimum 3 to 5 years of experience in AI solutions deployment in enterprise environments
  • Understanding of how LLMs (GPT-4, Gemini, Claude, Llama, Mistral AI) work, their capabilities, and limitations
  • Familiarity with model architecture, tokenization, context windows, and prompt formatting
  • Crafting effective prompts for various tasks (text generation, summarization, Q&A, code generation, images and sound manipulations)
  • Techniques for prompt chaining, few-shot and zero-shot learning, and multi-turn conversations
  • Knowledge of prompt templates, system instructions, and role-based prompting
  • Understanding the concept of AI agents: autonomous entities that perceive, reason, and act to achieve goals
  • Familiarity with multi-agent systems, agent orchestration, and agentic workflows (e.g., using frameworks like Lang Chain, Crew AI, Auto Gen)
  • Ability to design, prompt, and coordinate groups of AI agents for collaborative or competitive tasks
  • Knowledge of agent communication, delegation, and task decomposition
Job Responsibility
Job Responsibility
  • Develop, refine, and optimize prompts for LLMs (GPT-4, Gemini, Claude, Llama, Mistral) to support a variety of tasks such as text generation, summarization, Q&A, and code generation
  • Design and implement prompt strategies for multi-turn conversations, prompt chaining, and role-based instructions
  • Build and coordinate groups of AI agents (multi-agent systems) for collaborative or competitive tasks using frameworks such as Lang Chain, Crew AI, or Auto Gen
  • Upgrade existing prompts while releases and solutions set evolve
  • Evaluate and improve the effectiveness of prompts and agent workflows through iterative testing, A/B experimentation, and performance analysis
  • Collaborate with cross-functional teams (developers, data scientists, product managers) to integrate prompt engineering and agentic workflows into enterprise solutions
  • Ensure compliance with data privacy, security, and regulatory standards (GDPR, NIS2) in all prompt and agent designs
  • Document prompt strategies, agent architectures, and best practices for internal knowledge sharing and training
  • Ensure provided solutions are ready for production, documented and monitored ensuring consistent delivery and quality
What we offer
What we offer
  • Full access to foreign language learning platform
  • Personalized access to tech learning platforms
  • Tailored workshops and trainings to sustain your growth
  • Medical Insurance
  • Meal tickets
  • Monthly budget to allocate on flexible benefit platform
  • Access to 7 Card services
  • Wellbeing activities and gatherings
  • Fulltime
Read More
Arrow Right

AI Engineer

AI Engineer position at Inetum, a European leader in digital services, focusing ...
Location
Location
Romania , Bucharest
Salary
Salary:
Not provided
https://www.inetum.com Logo
Inetum
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Minimum 3 to 5 years of experience in AI solutions deployment in enterprise environments
  • Understanding of how LLMs (GPT-4, Gemini, Claude, Llama, Mistral AI) work, their capabilities, and limitations
  • Familiarity with model architecture, tokenization, context windows, and prompt formatting
  • Crafting effective prompts for various tasks (text generation, summarization, Q&A, code generation, images and sound manipulations)
  • Techniques for prompt chaining, few-shot and zero-shot learning, and multi-turn conversations
  • Knowledge of prompt templates, system instructions, and role-based prompting
  • Understanding the concept of AI agents: autonomous entities that perceive, reason, and act to achieve goals
  • Familiarity with multi-agent systems, agent orchestration, and agentic workflows (e.g., using frameworks like Lang Chain, Crew AI, Auto Gen)
  • Ability to design, prompt, and coordinate groups of AI agents for collaborative or competitive tasks
  • Knowledge of agent communication, delegation, and task decomposition
Job Responsibility
Job Responsibility
  • Develop, refine, and optimize prompts for LLMs (GPT-4, Gemini, Claude, Llama, Mistral) to support a variety of tasks such as text generation, summarization, Q&A, and code generation
  • Design and implement prompt strategies for multi-turn conversations, prompt chaining, and role-based instructions
  • Build and coordinate groups of AI agents (multi-agent systems) for collaborative or competitive tasks using frameworks such as Lang Chain, Crew AI, or Auto Gen
  • Upgrade existing prompts while releases and solutions set evolve
  • Evaluate and improve the effectiveness of prompts and agent workflows through iterative testing, A/B experimentation, and performance analysis
  • Collaborate with cross-functional teams (developers, data scientists, product managers) to integrate prompt engineering and agentic workflows into enterprise solutions
  • Ensure compliance with data privacy, security, and regulatory standards (GDPR, NIS2) in all prompt and agent designs
  • Document prompt strategies, agent architectures, and best practices for internal knowledge sharing and training
  • Ensure provided solutions are ready for production, documented and monitored ensuring consistent delivery and quality
What we offer
What we offer
  • Full access to foreign language learning platform
  • Personalized access to tech learning platforms
  • Tailored workshops and trainings to sustain your growth
  • Medical Insurance
  • Meal tickets
  • Monthly budget to allocate on flexible benefit platform
  • Access to 7 Card services
  • Wellbeing activities and gatherings
  • Fulltime
Read More
Arrow Right

Senior Platform Engineer, AI Evaluation

We’re looking for an AI Platform Engineer to evolve and extend our internal eval...
Location
Location
United States , Mountain View
Salary
Salary:
137871.00 - 172339.00 USD / Year
khanacademy.org Logo
Khan Academy
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor’s or Master’s degree in Computer Science, Data Engineering, or a related field
  • 5 years of Software Engineering experience with 2+ of those years working on the evaluation of generative AI systems
  • Strong programming skills in Go, Python, SQL, and at least one data pipeline framework (e.g., Airflow, Dagster, Prefect)
  • Familiarity with the architecture of large language models and their industry-standard APIs
Job Responsibility
Job Responsibility
  • Evolve and extend our internal evaluation framework for assessing the quality of our AI-driven experiences
  • Work closely with ML data engineers and platform developers to help internal teams adopt an eval-driven development process incorporating offline benchmark tests and online experiments
  • Gather internal requirements, getting buy-in for changes, and then developing documentation and training materials
What we offer
What we offer
  • Competitive salaries
  • Ample paid time off as needed
  • 8 pre-scheduled Wellness Days in 2026
  • Remote-first culture
  • Generous parental leave
  • 401(k) + 4% matching
  • Comprehensive insurance, including medical, dental, vision, and life
  • Fulltime
Read More
Arrow Right