Live

ML Optimization Engineer - Oslo

Squarehead Technology•Posted 2 weeks ago

Job Description

Squarehead Technology seeks an ML Optimization Engineer in Oslo to enhance machine learning models for spatial audio. Focus on compression techniques like quantization, pruning, and distillation using PyTorch, TensorFlow, and CUDA. Collaborate with scientists to integrate optimized models into production, improving speed and efficiency for advanced audio applications.

Required Skills

Deep learning internals with PyTorch, TensorFlow, or JAX including graph editing and custom operators Low-level programming in C++ and GPU acceleration with CUDA or Vulkan Compute Shaders Mixed-precision computation and memory-footprint optimization Digital signal processing theory and implementation Profiling and benchmarking for performance analysis Setting up regression tests for accuracy and performance

Click on skills to find similar jobs requiring these technologies

Requirements

Implement model compression techniques like PTQ, QAT, pruning, and knowledge distillation
Integrate and fine-tune models on runtimes such as TensorRT, ONNX Runtime, or CoreML
Use profiling tools like Nsight or TensorBoard to identify bottlenecks
Optimize DSP algorithms for audio and signal-processing pipelines
Collaborate with ML Scientists for end-to-end model integration and deployment
Proficiency in deep learning frameworks and low-level programming
Knowledge of DSP and audio fundamentals
Performance optimization under resource constraints