v1.4.1

Lux v1.4.1

Diff since v1.4.0

Merged pull requests:

• Update exporting_to_jax.md (#1107) (@wsmoses)
• CompatHelper: bump compat for LossFunctions in [weakdeps] to 1, (keep existing compat) (#1108) (@github-actions[bot])
• Add TrainState docstring with Optimisers API (#1110) (@abhro)
• Fix markdown list in docstring (#1111) (@abhro)
• chore: bump crate-ci/typos from 1.27.3 to 1.28.1 (#1113) (@dependabot[bot])
• fix: handle debug leafs with dispatch (#1115) (@avik-pal)
• test: allow the latest AMDGPU to be installed (#1116) (@avik-pal)
• test: add unsafe_free to skip list (#1117) (@avik-pal)
• fix: use the correct dispatches for device overloads (#1118) (@avik-pal)
• test: try fixing enzyme test (#1119) (@avik-pal)
• ci(github-actions): use julia-actions/cache (#1122) (@avik-pal)
• test: re-enable flux testing (#1123) (@avik-pal)
• chore: bump minimum Reactant version (#1125) (@avik-pal)
• fix: try fixing cuda install in tests (#1126) (@avik-pal)

Closed issues:

• Enzyme 0.13 fails with batched matrix multiply (#1024)
• getkeypath and layer_map not fully working with model with Parallel layers (#1068)
• Re-enable Flux compatibility testing (#1070)
• [AMDGPU CI] Circular dependencies disabling precompilation (#1095)
• LuxTestUtils.Constant clashes with DifferentiationInterface.Constant (#1103)
• Error in trying to use Optimization.jl for LSTM training based on Lux.jl (#1114)
• Documentation Build Stalls (#1120)
• CUDA Test CI is broken (#1121)