soraros

The `wrap/unwrap` names alone are not very infomatie, but `tree_math.wrap/unwrap` are. Some of the names listed above even look redundant in the quantified form.

It does reproduce on my MacBook, with jaxlib 0.3.15, and I think that's where the problem lies. ```python jax.print_environment_info() # jax: 0.3.18 # jaxlib: 0.3.15 # numpy: 1.23.3 # python:...

Mine is also an intel MacBook. And the problem persists with a locally built `jaxlib` from HEAD. ```python In [1]: import jax.numpy as jnp ...: import jax ...: ...: jax.config.update('jax_enable_x64',...

@hawkinsp Here you go. ``` machdep.cpu.mwait.linesize_min: 64 machdep.cpu.mwait.linesize_max: 64 machdep.cpu.mwait.extensions: 3 machdep.cpu.mwait.sub_Cstates: 286531872 machdep.cpu.thermal.sensor: 1 machdep.cpu.thermal.dynamic_acceleration: 1 machdep.cpu.thermal.invariant_APIC_timer: 1 machdep.cpu.thermal.thresholds: 2 machdep.cpu.thermal.ACNT_MCNT: 1 machdep.cpu.thermal.core_power_limits: 1 machdep.cpu.thermal.fine_grain_clock_mod: 1 machdep.cpu.thermal.package_thermal_intr: 1 machdep.cpu.thermal.hardware_feedback:...

@hawkinsp Here you go again. [dump.zip](https://github.com/google/jax/files/9637673/dump.zip)

@Mogball Regardless of how the internal works, I'd still expect this code to run. Because from a users perspective, `math.bit` just look like any other module. If they definitely can't...

@abduld Thanks for the pointer. I'm familiar with this implementation yet didn't know the existence of this function in the stdlib.

I'm against adding this as currently there is no way to type `divmod` properly (similar to how we don't have `min`/`max`). You can call `__divmod__` directly on `datetime` as a...

> we already have min and max, it only works with Int and SIMD at the moment Exactly: not as "builtin" functions like in Python. I think adding a `divmod`...

If by array, you meant array with fixed length, then `StaticTuple` is just that with a horrible name.