https://eli.thegreenplace.net/2025/reproducing-word2vec-with-jax/ · April 5, 2025