Add jax lowering to cast f64 to f32 for aten ops add, sub, lt, mul, where.#785

copybara-service[bot] wants to merge 1 commit intomaingoogle-ai-edge/ai-edge-torch:mainfrom

test_792393759google-ai-edge/ai-edge-torch:test_792393759Copy head branch name to clipboard