scvi-tools
scvi-tools copied to clipboard
Jax analog for FCLayers
Mentioned by @jjhong922 in #1608 - need a Jax equivalent of the PyTorch FCLayers to reduce redundancy in FlaxEncoder/Decoder
Mentioned by @jjhong922 in #1608 - need a Jax equivalent of the PyTorch FCLayers to reduce redundancy in FlaxEncoder/Decoder