`jax.numpy.fix` brings different results with `numpy.fix`

[apiqwe]

Description

I found that jax.numpy.fix brings different results with numpy.fix in following case.
I think this may be a bug in JAX.

import numpy as np
import jax.numpy as jnp

print(np.fix(-1000000.987))
print(jnp.fix(-1000000.987))

Output:

-1000000.0
-1000001.0

System info (python version, jaxlib version, accelerator, etc.)

jax:    0.6.0
jaxlib: 0.6.0
numpy:  2.2.3
python: 3.10.12 (main, Feb  4 2025, 14:57:36) [GCC 11.4.0]
device info: NVIDIA GeForce RTX 4090-1, 1 local devices"
process_count: 1
platform: uname_result(system='Linux', node='4db45dc420f8', release='6.11.0-25-generic', version='#25~24.04.1-Ubuntu SMP PREEMPT_DYNAMIC Tue Apr 15 17:20:50 UTC 2', machine='x86_64')


$ nvidia-smi
Wed May  7 16:26:08 2025       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.230.02             Driver Version: 535.230.02   CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 4090        Off | 00000000:01:00.0 Off |                  Off |
| 36%   32C    P2              36W / 450W |    407MiB / 24564MiB |      1%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
+---------------------------------------------------------------------------------------+

[dfm]

Thanks! I think this is actually working as intended. Unlike numpy, JAX operates in float32 by default. The following:

print(np.fix(np.float32(-1000000.987)))
print(jnp.fix(-1000000.987))

outputs

-1000001.0
-1000001.0

(because np.float32(-1000000.987) is rounded to -1000001.0). Enabling 64 bit types in JAX also prints consistent results with numpy:

jax.config.update("jax_enable_x64", True)
print(np.fix(-1000000.987))
print(jnp.fix(-1000000.987))

-1000000.0
-1000000.0