A collection (in progress) of example high-performance large language model implementations, written with JAX.
Current contents include:
Repository from Github https://github.commattjj/jax-llm-examples
A collection (in progress) of example high-performance large language model implementations, written with JAX.
Current contents include:
Apache License 2.0