Tags: google-deepmind/tf2jax

Bump version to 0.3.8

PiperOrigin-RevId: 797708508

Migrate from jax.core to jax.extend.core for several deprecated symbols

A number of symbols from jax.core are deprecated as of recent JAX releases; some of them are newly available in jax.extend.core.

PiperOrigin-RevId: 706738825

update pypi-publish workflow.

PiperOrigin-RevId: 619127773

Add jnp.bool_ in tf2jax as ir.IntegerType

PiperOrigin-RevId: 551197166

Allow variable k type and output index type in topk.

This is to help support smaller model sizes for some
tflite models. Requires expanding the TF API to allow
specification of the type, `index_type`.
The change is backward-compatible.

Some changes in XLA were required to support the new
API.

PiperOrigin-RevId: 523499712

Bump version to 0.3.3

PiperOrigin-RevId: 508716397

Bump version to 0.3.2

PiperOrigin-RevId: 503941229

Bump version to 0.3.1

PiperOrigin-RevId: 464782574

Bump version to 0.3.0

PiperOrigin-RevId: 456236286

Bump version to 0.2.1

PiperOrigin-RevId: 446991840