google-deepmind / ferminet

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations
Apache License 2.0
740 stars 129 forks source link

Fix code comment errors in the Jax implementation. #40

Closed shishaochen closed 2 years ago

shishaochen commented 2 years ago

For better readability, we'd better correct output shapes of these functions:

jsspencer commented 2 years ago

Thanks! We'd picked up one of these already but not the other.