- librarymoduleabout
- nxNxN-dimensional arrays as values on device sets.
- nx.arrayNx_arrayDtypes, layouts and arrays over device buffers.
- nx.cpuNx_cpuKernels for host memory.
- nx.ioNx_ioArrays to and from files.
- nx.kernelNx_kernelKernels: what they compute and the contract they implement.
- nx.quantNx_quantQuantised weights.
- nx.raggedNx_raggedRagged arrays.
- nx.topNx_topThe toplevel printer for arrays. #require "nx.top" installs Nx.pp, so the toplevel and utop show an array's values.
- nx.wideNx_wideDouble-word numbers.