Logo NXAI

Senior AI/ML Engineer

New

Job

  • Level
    Senior
  • Job Field
    Software, Data
  • Employment Type
    Full Time
  • Contract Type
    Permanent employment
  • Location
    Linz
  • Working Model
    Onsite
  • Job Summary

    In this role, you will develop AI models, optimize xLSTM/ViL models for deployment, and work on data handling, performance profiling, and deep learning pipeline optimization.

    Job Technologies

    Your role in the team

    At NXAI, we're always excited to meet forward-thinking minds eager to shape the future of AI.

    We work at the cutting edge of technology with a team of technical experts to continue creating excellent solutions.

    As Senior AI/ML Engineer, you will play a key role in building the bridge between research and production, ensuring xLSTM/ViL models are optimized, portable and ready for deployment.

    Your areas of responsibility include:

    • Deep Learning & Model Development
    • Model Optimization & Deployment
    • Data Handling & Preprocessing
    • Performance Profiling & Debugging

    This text has been machine translated. Show original

    Our expectations of you

    Qualifications

    You may be a good fit if you have:

    • Strong PyTorch skills for writing, debugging and optimizing models.
    • Profound understanding of transformers, LSTMs and attention mechanisms.
    • Time Series Forecasting expertise for long-sequence modeling.
    • Knowledge of converting models to ONNX format for deployment, TensorRT or MIGraphX for efficient inference, and TorchScript for exporting optimized models.
    • Core skills in software engineering and its best practices: e.g., structured, modular Python code for ML pipelines
    • Understanding of NVIDIA Nsight or PyTorch Profiler for speed/memory optimizations.
    • Proficiency in applying memory-efficient training strategies (mixed precision, gradient checkpointing).
    • Proven ability to provide technical guidance and act as a lead on complex projects.
    • You enjoy mentoring junior engineers and shaping the technical direction without losing yourself in PM-focused tasks.

    Experience

    • At least 5 years of experience or exceptional expertise with a strong track record, ideally in an industrial environment.
    • Experience with time-series data pipelines (Dask, Pandas, PySpark) and image preprocessing (OpenCV, torchvision).
    • Deep interest or experience at the intersection of Time Series, Computer Vision, and Robotics (e.g., path planning, sequence modeling, and sensor fusion).

    This text has been machine translated. Show original

    Job Locations

    Map of company locations
    • Location Linz

      Oberösterreich

      Austria

    Topics that you deal with on the job

    This is your employer

    NXAI

    NXAI

    Based in Linz, NXAI is a leading force in Artificial Intelligence research with a global perspective. NXAI's central mission is to translate research findings and technological advancements into market-relevant industrial applications.

    Description

  • Company Type
    Established Company
  • Working Model
    Onsite
  • Industry
    Internet, IT, Telecommunication
  • Logo NXAI

    Senior AI/ML Engineer

    Location
    Linz
    Working Model
    Onsite
    Diversity
    Open for all genders

    More Jobs