jax

e9c29d2f - Track platform on AbstractDevice. We already track device_kind and platform is a coarser version i.e. we won't get new trace time cache misses (you can have multiple device kinds for 1 platform).

Track platform on AbstractDevice. We already track device_kind and platform is a coarser version i.e. we won't get new trace time cache misses (you can have multiple device kinds for 1 platform). This allows `.lower()` to pass when `use_abstract_mesh` is used at the top level without requiring `.lower(lowering_platforms=...)`. PiperOrigin-RevId: 914481785