h2o4gpu
h2o4gpu copied to clipboard
Use less memory for prediction
Ideally it should running also on GPU. But even without it naive implementation https://github.com/h2oai/h2o4gpu/blob/master/src/interface_py/h2o4gpu/solvers/factorization.py#L309-L311 can be optimized to use less memory.