FluxML/Flux.jl

Pull Requests Commits

Reactant trainstep!: single-forward aux, host-side outputs, cache warning

CarloLucibello committed 63 days ago

f7969ea6

train_step! -> trainstep!

CarloLucibello committed 63 days ago

067e83d5

CarloLucibello committed 63 days ago

2dda675e

Support auxiliary loss outputs in train_step! and train! (#2711)

CarloLucibello committed 63 days ago

Verified c234ff91

add padding_idx to Embedding (#2708)

CarloLucibello committed 63 days ago

Verified 0170e9e2

Support auxiliary outputs in Enzyme `withgradient` (#2710)

CarloLucibello committed 63 days ago

Verified 0f7d6f33

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

CarloLucibello committed 64 days ago

Verified d744107a

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

CarloLucibello committed 64 days ago

Verified 2f03a1b0

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

CarloLucibello committed 64 days ago

Verified f5c123d8

cleanup repo (#2705)

CarloLucibello committed 65 days ago

Verified 10ee4907

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

CarloLucibello committed 65 days ago

Verified 0267f613

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

CarloLucibello committed 65 days ago

Verified 740f744a

Delegate normalization layers to NNlib functional operators (#2701)

CarloLucibello committed 65 days ago

Verified ee4ec090

Add bf16 precision helper (#2698)

CarloLucibello committed 66 days ago

Verified 1e78c106

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

CarloLucibello committed 66 days ago

Verified 7078bc8f

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

CarloLucibello committed 66 days ago

Verified bad1d314

Adaptive GPU memory management in train! (#2697)

CarloLucibello committed 66 days ago

Verified ca6e8c06

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

CarloLucibello committed 68 days ago

Verified f4a72bcd

Fix check_equal_leaves so gradient comparisons actually run (#2693)

CarloLucibello committed 79 days ago

Verified 780af631

unbreak some metal tests (#2692)

CarloLucibello committed 80 days ago

Verified d618f1ff

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

jeis4wpi committed 80 days ago

Verified d099c22c

Update default for FLUX_TEST_ENZYME (#2690)

wsmoses committed 87 days ago

Verified 38e189dd

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

dependabot[bot] committed 92 days ago

Verified 4d504e09

unbreak enzyme tests (#2688)

CarloLucibello committed 106 days ago

Verified 57e29baf

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

maleadt committed 106 days ago

Verified 79013391

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

dependabot[bot] committed 106 days ago

Verified d8646923

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

dependabot[bot] committed 115 days ago

Verified 5cdb7c57

Support auxiliary outputs in withgradient for Mooncake (#2682)

CarloLucibello committed 122 days ago

Verified f3d36846

Update AGENTS.md extension table (#2678)

CarloLucibello committed 123 days ago

Verified 4c95241c

Update ecosystem.md (#2674)

msainsburydale committed 162 days ago

Verified c2b51692

Older