docs: use more of Lux official API for training and inference#438

yebai merged 1 commit intoJuliaGaussianProcesses:copilot/fix-426JuliaGaussianProcesses/AbstractGPs.jl:copilot/fix-426from

avik-pal:ap/lux_improvementsavik-pal/AbstractGPs.jl:ap/lux_improvementsCopy head branch name to clipboard