Lux: Explicit Parameterization of Deep Neural Networks in Julia

WeightInitializers WeightInitializers-v1.3.0 Diff since WeightInitializers-v1.2.2 Merged pull requests: feat: precompile common workloads (#1485) (@avik-pal) feat: batched_jacobian for Reactant (#1507) (@avik-pal) refactor: use EnzymeRules.@easy_rule in Lux.jl (#1542) (@avik-pal) test: use finite differencing for gradient testing for Reactant (#1545) (@avik-pal) feat: more informative error on constructing trainstate with compiled function (#1547) (@avik-pal) fix: minor reactant stuff + docs build (#1548) (@avik-pal) feat: use a caching allocator for GPUArrays workflows (#1549) (@avik-pal) Avoid reconstruction in Internal.unsafe_free! (#1550) (@AntonOresten) test: update tests for enzyme (#1552) (@avik-pal) fix: update how the error message looks (#1553) (@avik-pal) Use |> for moving data to devices (#1559) (@abhro) feat: return sequence properly + checkpointing + mincut (#1561) (@avik-pal) fix(LuxLib): avoid extra copy if input and output are aliased (#1562) (@avik-pal) ci: use dependabot for updating compat entries (#1563) (@avik-pal) chore: bump actions/checkout from 5 to 6 (#1564) (@dependabot[bot]) chore: bump crate-ci/typos from 1.39.0 to 1.39.2 (#1565) (@dependabot[bot]) Put plot labels within plotting directives (#1566) (@abhro) Remove unnecessary begin...end markers (#1567) (@abhro) ci(docs): update cpu builds to use default gh actions (#1569) (@avik-pal) Indent code blocks to make it part of ordered list (#1570) (@abhro) Fix @example blocks in gpu_management.md (#1571) (@abhro) chore: bump actions/download-artifact from 5 to 6 (#1575) (@dependabot[bot]) ci: fix download path for cuda ci (#1576) (@avik-pal) chore: bump crate-ci/typos from 1.39.2 to 1.40.0 (#1578) (@dependabot[bot]) test: Metal test now works (#1581) (@avik-pal) Add AbstractChar array support to MLDataDevices (#1582) (@Copilot) test: streamline installing packages in tests (#1583) (@avik-pal) Mark LuxCore imports as public using @public (#1585) (@Copilot) Fix isbits type support for GPU device transfer (#1587) (@Copilot) Add OpenCL support to MLDataDevices (#1590) (@VarLad) docs: Recommend Reactant for AMD GPU support and add Tenstorrent backend (#1593) (@Copilot) docs: fix docs build (#1594) (@avik-pal) ci(docs): run all docs on cpu runners (#1595) (@avik-pal) chore: bump actions/upload-artifact from 5 to 6 (#1598) (@dependabot[bot]) chore: bump actions/download-artifact from 6 to 7 (#1599) (@dependabot[bot]) chore: bump peter-evans/create-pull-request from 7 to 8 (#1600) (@dependabot[bot]) feat: proper annotations for xprof (#1603) (@avik-pal) fix: forwarddiff support for gpu arrays (#1605) (@avik-pal) docs: use new API to export jax models (#1606) (@avik-pal) chore: update NPZ requirement to 0.4.3 in /docs (#1607) (@dependabot[bot]) chore: update MLUtils requirement to 0.4.8 in /docs (#1608) (@dependabot[bot]) ci: reactant don't preallocate on CI (#1611) (@avik-pal) ci: fix how downgrade ci works (#1612) (@avik-pal) feat: make forwarddiff a weak dependency for luxlib (#1613) (@avik-pal) perf: more benchmarking results (#1614) (@avik-pal) chore: bump crate-ci/typos from 1.40.0 to 1.41.0 (#1616) (@dependabot[bot]) fix: sparse arrays support (#1617) (@avik-pal) chore: rename extensions in LuxCore (#1618) (@avik-pal) chore: rename extensions in LuxLib (#1619) (@avik-pal) chore: rename extensions in MLDataDevices & WI (#1620) (@avik-pal) Closed issues: Proper Sparse Interfaces (#1014) Add simple tests for other accelerators (#686) DifferentiationInterface testing (#769) Automatically cache allocations for JuliaGPU workloads (#1527) Relax ForwardDiff version bound? (#1530) Embedding Layer results in scalar indexing with Reactant? (#1546) Enzyme Cache Invalidation Failure with v1.10 (#1551) OneHotArrays + Reactant with cross entropy loss (#1556) Overhead of convolution on AMD GPU (#1557) LuxTestUtils not re-exported (#1579) [MLDataDevices] failure at transfering non numerical array (#1580) Mark LuxCore imports in Lux as public (#1584) [MLDataDevices] broken support for isbits types movement (#1586) Update exporting to jax example to directly use new export functionality (#1588) Failed to precompile LuxLossFunctionsExt (#1591) Update AMD support documentation to recommend using Reactant (#1592) ERROR: KeyError: key "ReactantCore" not found (#1596) TagBot: Manual intervention needed for releases (#1604) What is wrong with my custom layer? (#1610) [MLDataDevices] AbstractCuSparseArray not defined in CUDA.jl v5.9.6 (#1615)