# Does Lux work with FluxTraining?

**URL:** https://discourse.julialang.org/t/does-lux-work-with-fluxtraining/108591
**Category:** Machine Learning
**Tags:** fluxtraining
**Created:** [January 9, 2024, 9:33pm UTC](https://discourse.julialang.org/t/does-lux-work-with-fluxtraining/108591 "2024-01-09T21:33:38Z")
**Posts on this page:** 1
**Page:** 1

<div class="post-metadata">

### Author: ![neuromancer](https://avatars.discourse-cdn.com/v4/letter/n/46a35a/32.png) [@neuromancer](https://discourse.julialang.org/u/neuromancer)
#### Post date: [January 9, 2024, 9:33pm UTC](https://discourse.julialang.org/t/does-lux-work-with-fluxtraining/108591/1 "2024-01-09T21:33:38Z")

</div>

I’m thinking about switching from Flux to Lux, since I really need a WeightNorm layer and I am struggling to write one for myself. I am heavily entrenched into using FluxTraining. Does anyone know if Lux works with FluxTraining (or if there’s a WeightNorm implementation for Flux)?