[jax2tf] Remove dot_general limitation due to XLA fixing crashing bug
[jax2tf] Remove dot_general limitation due to XLA fixing crashing bug

PiperOrigin-RevId: 373375341

google/jax,google/jax,google/jax,google/jax