AI Frameworks Engineer

Great Sky • Palo Alto, CA, US • published 2026-09-23 05:43:16.016323

Description

**About Us** ------------ Great Sky is a technology startup based with a mission to rebuild AI from first principles. We are pursuing neuroscience\-inspired hardware and algorithms that overcome the greatest challenges in scaling AI systems. **The Role** ------------ We are looking for an AI Frameworks Engineer to help build the core software layer that connects our research ideas to working, testable, scalable AI systems. This is a software engineering role focused on the internal AI framework: the abstractions, APIs, graph execution paths, training strategies, validation systems, and test infrastructure that make our research stack usable and reliable. **Key Responsibilities** ------------------------ * Design, implement, and maintain core AI framework abstractions for defining, training, validating, and executing novel graph\-structured models. * Build clean Python APIs for training workflows, including loss objects, model runtime bundles, optimizer/training strategy interfaces, and more. * Develop and maintain JAX and PyTorch infrastructure for model execution, customizable training protocols, numerical parity tests, and reusable forward\-pass utilities. * Help design validatio...

Get jobs like this daily in Telegram: @careeri_bot