flax

82daae36 - Merge pull request #2364 from cgarciae:fix-transpose-conv