jax

Fix v_pages quantization and adjust test params for ROCm compatibility

#559

Merged