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 (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. Yourareas of responsibility include:
- Deep Learning & Model Development
- Model Optimization & Deployment
- Data Handling & Preprocessing
- Performance Profiling & Debugging
You may be a good fit if you have:
- At least 5 years of experience or exceptional expertise with a strong track record, ideally in an industrial environment.
- 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
- 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).
- Deep interest or experience at the intersection of Time Series, Computer Vision, and Robotics (e.g., path planning, sequence modeling, and sensor fusion).
- 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.
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 shaping the future of AI.
Tags & focus areas
Used for matching and alerts on DevFound Fulltime Ai Engineer Machine Learning Deep Learning Pytorch Ai