Hongkai Zheng

I tried to sample images using the code and checkpoint from [diffusion_distillation](https://github.com/google-research/google-research/blob/3b61da1d73543c374fc0564943e237db151af452/diffusion_distillation/README.md) to reproduce their results in this paper [PROGRESSIVE DISTILLATION FOR FAST SAMPLING OF DIFFUSION MODELS](https://arxiv.org/abs/2202.00512). However, the FID...

Inconsistent results of Flax Dense and Pytorch Linear

5

Hi everyone, I'm reading the Flax document about [converting Pytorch models to Flax](https://flax.readthedocs.io/en/latest/guides/convert_pytorch_to_flax.html) and trying to test some code myself. But I observed inconsistency between Flax Dense and Pytorch linear....