---
date: '2024-03-04'
description: PTX equivalent for XLA
id: PJRT
modified: 2026-06-05 15:08:22 GMT-04:00
tags:
  - ml
title: PJRT
created: '2024-03-04'
published: '2024-03-04'
pageLayout: default
slug: thoughts/PJRT
permalink: https://aarnphm.xyz/thoughts/PJRT.md
generator:
  quartz: v4.6.0
  hostedProvider: Cloudflare
  baseUrl: aarnphm.xyz
full: https://aarnphm.xyz/llms-full.txt
---
Blog [post](https://opensource.googleblog.com/2023/05/pjrt-simplifying-ml-hardware-and-framework-integration.html) and [source](https://github.com/openxla/xla/tree/main/xla/pjrt)

Lower stack layer for framework and hardware communication.

As abstraction to transpile to different hardware targets: [[thoughts/TPU|TPU]], [[thoughts/GPU programming|GPU]]