[JAX] Keep CPU host callbacks alive via IFRT, rather than by attachin… #10

	# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.
	# ============================================================================
	name: Rollback Notification
	on:
	push:
	branches:
	- main

	jobs:
	rollback-notification:
	if: contains(github.event.head_commit.message, 'revert')
	permissions:
	pull-requests: write
	runs-on: ubuntu-22.04
	defaults:
	run:
	shell: bash
	env:
	GH_TOKEN: ${{ github.token }}
	HEAD_COMMIT_MESSAGE: ${{ github.event.head_commit.message }}
	timeout-minutes: 1
	steps:
	- name: "Checking out repository"
	uses: actions/checkout@e2f20e631ae6d7dd3b768f56a5d2af784dd54791 # v2.5.0
	- name: "Check if PR was rolled back"
	run: python3 .github/workflows/rollback_notification.py

Provide feedback