# Lux (And Flux), "parallel" Network Input. When Input is flat, Zygote gradient works, when input is not flat it doesn't

**URL:** https://discourse.julialang.org/t/lux-and-flux-parallel-network-input-when-input-is-flat-zygote-gradient-works-when-input-is-not-flat-it-doesnt/107763
**Category:** Machine Learning
**Tags:** flux, zygote, lux
**Created:** [December 18, 2023, 10:15am UTC](https://discourse.julialang.org/t/lux-and-flux-parallel-network-input-when-input-is-flat-zygote-gradient-works-when-input-is-not-flat-it-doesnt/107763 "2023-12-18T10:15:39Z")
**Posts on this page:** 1
**Showing post:** 9

<div class="post-metadata">

### Author: ![ToucheSir](https://sea2.discourse-cdn.com/julialang/user_avatar/discourse.julialang.org/touchesir/32/14411_2.png) [@ToucheSir](https://discourse.julialang.org/u/ToucheSir)
#### Post date: [January 30, 2024, 3:36pm UTC](https://discourse.julialang.org/t/lux-and-flux-parallel-network-input-when-input-is-flat-zygote-gradient-works-when-input-is-not-flat-it-doesnt/107763/9 "2024-01-30T15:36:07Z")

</div>

Per my last message, I would remove all uses of `params` and instead follow the docs on how to take gradients of models. If you’re still running into issues then a MWE and full stacktrace will be required, per [Please read: make it easier to help you](https://discourse.julialang.org/t/please-read-make-it-easier-to-help-you/14757).

---

_[View the full topic](https://discourse.julialang.org/t/lux-and-flux-parallel-network-input-when-input-is-flat-zygote-gradient-works-when-input-is-not-flat-it-doesnt/107763)._