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 Machine Learning Engineer (f/m/d), 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 DevelopmentModel Optimization & DeploymentData Handling & PreprocessingPerformance Profiling & DebuggingYou 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: structured, modular Python code for ML pipelines and basic C++ knowledge.Experience with time-series data pipelines (Dask, Pandas, PySpark) and image preprocessing (OpenCV, torchvision).Understanding of NVIDIA Nsight or PyTorch Profiler for speed/memory optimizations.Proficiency in applying memory-efficient training strategies (mixed precision, gradient checkpointing).What you’ll get:We offer you the opportunity to work in a key position at the forefront of a fast-growing company.Competitive salary in line with the market, plus a tailored benefits package.Benefit from a diverse and probably one of the best teams in the field of new evolving AI technology, where professional excellence is the guiding principle.The chance to make a real impact and shape the future of AI.Company:NXAIQualifications:Language requirements:Specific requirements:Educational level:Level of experience (years):Senior (5+ years of experience)Tagged as: Austria, Industry, Machine Learning, NLP
#J-18808-Ljbffr