FluxML/Flux.jl

Pull Requests Commits

Support auxiliary outputs in Enzyme `withgradient`

CarloLucibello committed 62 days ago

93d6298e

Add public `train_step!`, build `train!` on top of it (#2709)

CarloLucibello committed 62 days ago

Verified d744107a

Compile `train!` on Reactant devices (#2707)

CarloLucibello committed 62 days ago

Verified 2f03a1b0

perf/reactant: benchmark Adam (Optimisers 0.4.8), fix BatchNorm loss-probe artifact (#2706)

CarloLucibello committed 63 days ago

Verified f5c123d8

cleanup repo (#2705)

CarloLucibello committed 63 days ago

Verified 10ee4907

Add Zygote-vs-Reactant training benchmark (perf/reactant) (#2704)

CarloLucibello committed 63 days ago

Verified 0267f613

Mixed-precision f16/bf16 for BatchNorm/InstanceNorm/GroupNorm (#2700)

CarloLucibello committed 64 days ago

Verified 740f744a

Delegate normalization layers to NNlib functional operators (#2701)

CarloLucibello committed 64 days ago

Verified ee4ec090

Add bf16 precision helper (#2698)

CarloLucibello committed 64 days ago

Verified 1e78c106

Overhaul NEWS.md and document changelog upkeep (#2699)

CarloLucibello committed 64 days ago

Verified 7078bc8f

Add a ResNet-18 / Tiny-ImageNet-200 example (#2696)

CarloLucibello committed 65 days ago

Verified bad1d314

Adaptive GPU memory management in train! (#2697)

CarloLucibello committed 65 days ago

Verified ca6e8c06

Add option to disable the caching allocator in train! (#2695)

CarloLucibello committed 66 days ago

Verified f4a72bcd

Fix check_equal_leaves so gradient comparisons actually run (#2693)

CarloLucibello committed 77 days ago

Verified 780af631

unbreak some metal tests (#2692)

CarloLucibello committed 78 days ago

Verified d618f1ff

docs: fix typos and trim trailing whitespace (#2691)

jeis4wpi committed 78 days ago

Verified d099c22c

Update default for FLUX_TEST_ENZYME (#2690)

wsmoses committed 85 days ago

Verified 38e189dd

Bump dorny/paths-filter from 4.0.1 to 4.0.2 (#2689)

dependabot[bot] committed 91 days ago

Verified 4d504e09

unbreak enzyme tests (#2688)

CarloLucibello committed 104 days ago

Verified 57e29baf

Update Buildkite pipeline for the new JuliaGPU cluster (#2685)

maleadt committed 104 days ago

Verified 79013391

Bump actions/checkout from 6 to 7 (#2687)

dependabot[bot] committed 104 days ago

Verified d8646923

Bump codecov/codecov-action from 6 to 7 (#2683)

dependabot[bot] committed 113 days ago

Verified 5cdb7c57

Support auxiliary outputs in withgradient for Mooncake (#2682)

CarloLucibello committed 121 days ago

Verified f3d36846

Update AGENTS.md extension table (#2678)

CarloLucibello committed 121 days ago

Verified 4c95241c

Update ecosystem.md (#2674)

msainsburydale committed 160 days ago

Verified c2b51692

Bump julia-actions/setup-julia from 2 to 3 (#2673)

dependabot[bot] committed 168 days ago

Verified 7b2849a2

Bump version from 0.16.9 to 0.16.10

CarloLucibello committed 172 days ago

Verified a2c6e5c6

Bump codecov/codecov-action from 5 to 6 (#2668)

dependabot[bot] committed 174 days ago

Verified 9eed341e

CompatHelper: bump compat for cuDNN in [weakdeps] to 6, (keep existing compat) (#2671)

github-actions[bot] committed 174 days ago

Verified 818ad1a2

CompatHelper: bump compat for CUDA in [weakdeps] to 6, (keep existing compat) (#2672)

github-actions[bot] committed 174 days ago

Verified 7bf1fbb3

Older