jax

57975b18 - [XLA] Add a primitive for tagging individual XLA ops with frontend_attributes.