Thanks for sharing this! It's great to have a reference implementation written on java lang. With given original simplicity it's really easy to follow llama arc

Thanks for sharing this! It's great to have a reference implementation written on java lang. With given original simplicity it's really easy to follow llama architecture logic. Just in case if anyone interested in Python version, I spend some time on weekend and ported it to pure python -- https://github.com/tairov/llama2.py https://github.com/tairov/llama2.py I never knew that it would take about 500 lines of core part code to implement inference for such a cutting edge AI technology.