Pytorch Onnx Jobs (NOW HIRING)

Responsibilities : • Convert, optimize and deploy models for efficient inference using PyTorch, ONNX. • Work at the forefront of GenAI by understanding advanced algorithms (e.g. attention ...

Build and maintain model ingestion pipelines (e.g., PyTorch / ONNX → internal IR) * Implement graph transformations such as: * Operator decomposition and canonicalization * Shape inference and ...

Kubernetes, Ray, vLLM and PyTorch, onnx etc. * Strong programming skills in Python, C/C++, CUDA. * Excellent communication skills.

Build and maintain model ingestion pipelines (e.g., PyTorch / ONNX → internal IR) * Implement graph transformations such as: * Operator decomposition and canonicalization * Shape inference and ...

... PyTorch, ONNX, JAX Preferred : • 1+ years Python programming experience • Experience with different NN architectures: DNNs, CNNs, RNNs/LSTMs, GANs, LLMs, etc. • Experience with Graph ...

Guide model porting across frameworks and runtimes such as PyTorch, ONNX, and vendor-specific runtimes * Build prototypes and proofs of concept to reduce technical risk before engineering investment

Convert, optimize and deploy models for efficient inference using PyTorch, ONNX. * Work at the forefront of GenAI by understanding advanced algorithms (e.g. attention mechanisms, MoEs) and numerics ...

... PyTorch, ONNX and deployment/optimization of models is desired • Industry experience writing production-quality, performance-critical code, and maintaining large codebases is desired Company

PyTorch, ONNX/TensorRT export, and the tooling that keeps annotation, training, and deployment moving.* Partner with the edge team to quantize and prune models for on-device inference.## What you'll ...

... PyTorch, ONNX, and torch.compile, including model rewrites and graph-level transformations. • Design and implement fusion kernels using DSL based approaches (e.g., Triton), enabling fused ...

Experience working with one of the Deep Learning frameworks like TensorFlow, PyTorch, ONNX, JAX Preferred Qualifications * 1+ years Python programming experience * Experience with different NN ...

Experience in developing or using deep learning frameworks (e.g., TensorFlow, Kera's, PyTorch, ONNX, etc.) * Experience in optimization of deep learning/ML algorithms (e.g., Retraining, Quantization ...

... PyTorch, ONNX) and graph-level optimizations • Familiarity with ML hardware accelerators, memory hierarchies, and performance modeling • Solid C++ programming skills • Proven ability to lead ...

Experience deploying or optimizing neural network inference workloads using technologies such as PyTorch, ONNX, and TensorRT . * Experience with real-time embedded systems and handling large data ...

Guide model porting across frameworks and runtimes (e.g., PyTorch ONNX vendor specific runtimes) * Build prototypes and proof of concepts to reduce technical risk prior to full engineering investment ...

Experience deploying or optimizing neural network inference workloads using technologies such as PyTorch, ONNX, and TensorRT . * Experience with real-time embedded systems and handling large data ...

... PyTorch ONNX vendor specific runtimes) Build prototypes and proof of concepts to reduce technical risk prior to full engineering investment Qualifications Required Bachelor's or Master's degree in ...

Drive graph capture and deployment using PyTorch, ONNX, and torch.compile, including model rewrites and graph-level transformations. * Design and implement fusion kernels using DSL based approaches ...

Experience in ML frameworks (TensorFlow, PyTorch, ONNX) * Experience in micro architecture of SoC, interface subsystems, logic modules * Self-motivated problem-solver with an ability to work well in ...

Prior experience with AI frameworks and engines, such as TensorRT, PyTorch, ONNX, OpenVINO, vLLM, or TRT-LLM. * Knowledge of GPU memory management, cache management, or high-performance networking.