+ "source": "Besides JAX, it is possible to run PyFixest on the GPU via CuPy (linux and windows). Instead of applying the alternating projections algorithm to demean fixed effects, CuPy works with sparse matrices and the sparse LSMR solver (as is e.g. [available in scipy](https://docs.scipy.org/doc/scipy/reference/generated/scipy.sparse.linalg.lsmr.html)).\n\nThis strategy is amenable for GPU acceleration, and for problems where the standard demeaner struggles to converge, this strategy can lead to significant speedups if paired with a GPU.\n\n\n\nNote that for smaller and more well-behaved problems, running the alternating projections algorithm on the CPU via `numba` or `rust` usually seems to work better: \n\n\n\n**Benchmark Hardware Specifications:**\n- **CPU**: x86_64, 8 physical cores @ 3.2 GHz, 44 GB RAM\n- **GPU**: NVIDIA RTX A6000, 48 GB memory, Compute Capability 8.6"
0 commit comments