How to optimize JS code to speed up work?

Question

JavaScript

Good day.
The task is to calculate the values ​​​​of the matrix 52000 by 52000 on JS.
The value in each cell is the result of calculating the Cosine Proximity of 300 vectors.
To speed up the calculations, I use GPUJS:

gpu.addFunction(function mF(a, b) {
            return a * b;
        });
const f3 = gpu.createKernel(function(inp) {

            var a = 0
            var b = 0
            var c = 0
            for (var i = 0; i < 300; i++) {
                a += mF(inp[this.thread.y][i], inp[this.thread.x][i])
                b += mF(inp[this.thread.y][i], inp[this.thread.y][i])
                c += mF(inp[this.thread.x][i], inp[this.thread.x][i])
            }

            return a / (Math.pow(b, 0.5) * Math.pow(c, 0.5));

        }).setOutput([52000, 52000])

Can this be optimized?
On average, it takes 7ms to calculate all the cosine values ​​from one vector to all the others (36ms without a GPU), which is almost 6 minutes to calculate the entire matrix. I am a layman in JS and, most likely, I do not see obvious ways to optimize the code to speed up its work.
The script also dies safely when calculating the entire huge matrix, you have to count it in pieces of 1000 vectors.
Thanks

Answer 1

The script also dies safely when calculating the entire huge matrix, you have to count it in pieces of 1000 vectors.

Maybe it makes sense to send these pieces to SharedWorker

Answer 2

You can try WebAssembly

Answer 3

I have never dealt with GPUJS, but what catches your eye from general considerations is to cache indexes and be sure to replace pow 0.5 with sqrt (and with one, not two!).