FluxML/Flux.jl

Pull Requests Commits
Fix Dense with a zero-sized batch dimension (#2407)
CarloLucibello committed 57 days ago
b3be2f63
fix stack overflow when passing empty struct (#2670)
hardikxk committed 58 days ago
Verified 18f475ef
BatchNorm: error on a single value per channel in training (fixes #1992) (#2716)
CarloLucibello committed 60 days ago
Verified 3f2f828b
Bump version from 0.16.10 to 0.16.11
CarloLucibello committed 62 days ago
Verified 443ed1bc
Resolve AD backend once for `train!`/`trainstep!`; honour explicit `adtype` on Reactant (#2714)
CarloLucibello committed 62 days ago
Verified 6af22a1e
Reactant docs guide + `trainstep!` in the ResNet example & benchmark (#2713)
CarloLucibello committed 62 days ago
Verified c7b0217c
Reactant trainstep!: single-forward aux, host-side outputs, cache warning (#2712)
CarloLucibello committed 62 days ago
Verified cea47b23
train_step! -> trainstep!
CarloLucibello committed 63 days ago
067e83d5
cleanup
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 63 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 64 days ago
Verified 10ee4907
Add Zygote-vs-Reactant training benchmark (perf/reactant) (#2704)
CarloLucibello committed 64 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 65 days ago
Verified 1e78c106
Overhaul NEWS.md and document changelog upkeep (#2699)
CarloLucibello committed 65 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 67 days ago
Verified f4a72bcd
Fix check_equal_leaves so gradient comparisons actually run (#2693)
CarloLucibello committed 78 days ago
Verified 780af631
unbreak some metal tests (#2692)
CarloLucibello committed 79 days ago
Verified d618f1ff
docs: fix typos and trim trailing whitespace (#2691)
jeis4wpi committed 79 days ago
Verified d099c22c
Update default for FLUX_TEST_ENZYME (#2690)
wsmoses committed 86 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 105 days ago
Verified 57e29baf
Terms Of Use Privacy Policy Refund Policy Impressum