AI Model Developer

Ericsson GmbH; Austin (TX); On-site; USD 150,000 - 190,000; Full time

Job summary

Ericsson Inc. in Austin is seeking an experienced AI Model Developer to design, evaluate, optimize and deploy advanced ML models in a hands-on role.

You will work with researchers, software engineers, data scientists, and product teams to build transformer-based models and deploy them in production settings. The role requires 6+ years in industry, strong Python and ML workflows, and experience with PyTorch/JAX/TF.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, AI, Machine Learning, or a related field
• 6+ years of industry experience developing machine learning or AI solutions
• Solid experience with PyTorch, JAX, or TensorFlow
• Strong understanding of Transformer architectures, attention mechanisms, embeddings, and LLM fundamentals
• Proficiency in Python and modern ML development workflows
• Familiarity with model evaluation, experimentation, and performance analysis
• Experience with Git and collaborative software development practices
• Clear communication skills and ability to work effectively across teams

Responsibilities

• Design, develop, and optimize Transformer-based and other deep learning models
• Evaluate emerging AI architectures and recommend evidence-based improvements
• Build and maintain reference implementations using PyTorch or JAX
• Develop reproducible experiments and benchmark model accuracy, latency, and resource utilization
• Fine-tune foundation models for telecommunications and enterprise use cases
• Work with data scientists to prepare datasets and evaluation pipelines
• Partner with software engineering teams to integrate models into production environments
• Identify performance bottlenecks and recommend optimization strategies
• Document model designs, assumptions, and validation results clearly
• Support the growth of junior team members and contribute to shared engineering best practices

Skills

6+ years experience

PyTorch

JAX

TensorFlow

Python

Git

Communication

Transformer architectures

LLM fundamentals

Education

Bachelor’s or Master’s in CS/EE/AI

Tools

Hugging Face

ONNX

MLFlow

Ray

Job description

Overview

Grow with us! This is a hybrid position requiring onsite work three days a week at the Ericsson office in Austin, Texas. Relocation is not offered for this position. Ericsson Inc. does not sponsor U.S. work authorizations for this position (including initial and/or change of employer filings for H-1B, H-1B1, E-3, O-1, and TN visas). Ericsson does not hire F-1 students on CPT or EAD for this position.

About This Opportunity

Join a team building AI capabilities that power next-generation telecommunications solutions. Based in Austin, the AI Model Developer (JS5/JS6) works with researchers, software engineers, data scientists, and product teams to design, evaluate, optimize, and deploy advanced machine learning models. This is a hands-on technical role for an experienced professional with approximately 6+ years of industry experience who enjoys solving complex AI challenges and collaborating across disciplines.

Responsibilities

• Design, develop, and optimize Transformer-based and other deep learning models
• Evaluate emerging AI architectures and recommend evidence-based improvements
• Build and maintain reference implementations using PyTorch or JAX
• Develop reproducible experiments and benchmark model accuracy, latency, and resource utilization
• Fine-tune foundation models for telecommunications and enterprise use cases
• Work with data scientists to prepare datasets and evaluation pipelines
• Partner with software engineering teams to integrate models into production environments
• Identify performance bottlenecks and recommend optimization strategies
• Document model designs, assumptions, and validation results clearly
• Support the growth of junior team members and contribute to shared engineering best practices

The Skills You Bring

Required

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, AI, Machine Learning, or a related field
• 6+ years of industry experience developing machine learning or AI solutions
• Solid experience with PyTorch, JAX, or TensorFlow
• Strong understanding of Transformer architectures, attention mechanisms, embeddings, and LLM fundamentals
• Proficiency in Python and modern ML development workflows
• Familiarity with model evaluation, experimentation, and performance analysis
• Experience with Git and collaborative software development practices
• Clear communication skills and ability to work effectively across teams

Preferred

• Experience with Generative AI and Large Language Models
• Background applying AI to wireless communications, networking, or telecommunications
• Familiarity with Hugging Face, ONNX, MLFlow, Ray, or similar ML tooling
• Experience with distributed training or inference optimization
• Knowledge of MLOps and CI/CD practices for AI systems

What Happens Once You Apply

Click Here to find all you need to know about our typical hiring process.

EEO and Accommodations

Ericsson uses a merit-based hiring approach and is an Equal Opportunity employer. If you need assistance or to request an accommodation due to a disability, please contact [e-mail].

Disclaimer

The statements describe the general nature and level of work performed by employees in this position and are not an exhaustive list of responsibilities, duties, and skills required. Additional tasks may be assigned as needed.

Compensation and Benefits

Compensation and benefits information is provided separately. Ericsson offers a comprehensive benefits package including health, retirement, paid time off, and additional programs described by the company.