Lead AI/ML Engineer — Build Core Models (Hybrid, Austin)

Stelvio Inc.; Austin (TX); Hybrid; USD 180,000 - 260,000; Full time

Job summary

Austin-based Stelvio Inc. is seeking Principal/Staff-level AI/ML Engineers to take significant ownership of core model-development work. This highly autonomous role focuses on building, training and optimizing AI models rather than simple integration.

You will work closely with founders and the leadership team to design experiments, move models from early development through production deployment and shape the future of the company’s AI capabilities.

Qualifications

• Five+ years in AI/ML, research, or related software engineering.
• Built, trained or materially modified machine-learning models.
• Experience deploying models in production environments.
• Advanced Python development skills.
• Commercial experience with PyTorch, JAX, TensorFlow or comparable frameworks.
• Strong understanding of modern deep-learning and model-training techniques.
• Experience measuring and improving model performance.
• Ability to work independently within technically ambiguous environments.
• Experience delivering AI capabilities within a commercial or B2B setting.

Responsibilities

• Designing and training specialist machine-learning models.
• Building models from the ground up where the problem requires it.
• Fine-tuning and post-training foundation models using proprietary datasets.
• Owning experimentation, evaluation, benchmarking and model validation.
• Taking research concepts through to reliable production systems.
• Improving accuracy, latency, throughput, reproducibility and infrastructure efficiency.
• Optimizing model training and inference across GPU environments.
• Contributing to proprietary and potentially patentable AI technology.
• Helping define future model architecture and technical direction.
• Maintaining direct ownership of your work from initial concept through deployment.

Skills

Python

PyTorch

JAX

TensorFlow

Deep learning

Independent work

Production deployment

Model evaluation

Commercial AI

Tools

CUDA

Job description

Austin-based Stelvio Inc. is seeking Principal/Staff-level AI/ML Engineers to take significant ownership of core model-development work. This highly autonomous role focuses on building, training and optimizing AI models rather than simple integration.

You will work closely with founders and the leadership team to design experiments, move models from early development through production deployment and shape the future of the company’s AI capabilities.