Use axes instead of 1:size

[jishnub]

This generally makes it easier to read code, as one doesn't need to look up variable definitions from elsewhere.

[codecov]

Codecov Report

Attention: Patch coverage is 96.26168% with 4 lines in your changes missing coverage. Please review.

Project coverage is 84.08%. Comparing base (ce852af) to head (9bfcd22).
Report is 2 commits behind head on main.

Files with missing lines	Patch %	Lines
src/linalg.jl	92.72%	4 Missing ⚠️

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #602   +/-   ##
=======================================
  Coverage   84.08%   84.08%           
=======================================
  Files          12       12           
  Lines        9192     9192           
=======================================
  Hits         7729     7729           
  Misses       1463     1463           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.