Ayaka

Format of `epath.Path.stat()`: https://github.com/google/etils/blob/v1.9.2/etils/epath/stat_utils.py#L21-L39 Format of `pathlib.Path.stat()`: https://docs.python.org/3/library/os.html#os.stat_result Apart from the attribute name differences, the attribute that I am currently interested in is `pathlib.Path.stat().st_mtime_ns`, which represents mtime in nanoseconds. However,...

The documentation of [`pathlib.Path.touch`](https://docs.python.org/3/library/pathlib.html#pathlib.Path.touch): > Create a file at this given path. If mode is given, it is combined with the process’ umask value to determine the file mode and...

Failed to upload the new version to PyPI due to this error: ``` ERROR HTTPError: 400 Bad Request from https://upload.pypi.org/legacy/ Can't have direct dependency: tpu-info@ git+https://github.com/google/cloud-accelerator-diagnostics.git@6439522 892666f9a50ce3bfad7f9a12d4c7e136b#egg=tpu-info&subdirectory=tpu_info. See https://packaging.python.org/specifications/core-metadata for...

[Pallas] Fix integer array indexing

1

Fixes https://github.com/google/jax/issues/22783 Note: the code works now but still need some clean up

[Pallas TPU] Add lowering for `lax.erf_inv` 64 bit

4

Add lowering for `lax.erf_inv` 64 bit. This is a follow-up of #22282

[Pallas TPU] Wrong value when using `input_output_aliases` with multiple arrays

1

## Repro ```python import functools import jax from jax.experimental import pallas as pl import jax.numpy as jnp @functools.partial( pl.pallas_call, out_shape=( jax.ShapeDtypeStruct((2,), jnp.float32), jax.ShapeDtypeStruct((2,), jnp.float32), ), grid=1, input_output_aliases={0: 0, 1: 1},...

[Pallas TPU] Error when negating a boolean value

1

### Description ```python import functools import jax from jax.experimental import pallas as pl import jax.numpy as jnp @functools.partial( pl.pallas_call, out_shape=jax.ShapeDtypeStruct((2,), jnp.bool_), ) def kernel(x_ref, o_ref): o_ref[...] = jnp.logical_not(x_ref[...]) def main()...

### Description ```python import functools import jax from jax.experimental import pallas as pl from jax.experimental.pallas import tpu as pltpu import jax.numpy as jnp @functools.partial( pl.pallas_call, in_specs=( pl.BlockSpec(memory_space=pltpu.TPUMemorySpace.SMEM), pl.BlockSpec(memory_space=pltpu.TPUMemorySpace.SMEM), ), out_specs=pl.BlockSpec(memory_space=pltpu.TPUMemorySpace.SMEM),...

Add support for multiple languages, such as English and Japanese

《新潮汕字典》（第 2 版）p571 用例 https://github.com/user-attachments/assets/8de0da0d-5605-4581-9672-20de7c858122