Pallas: a JAX kernel language — JAX documentation - discu.eu

Linking pages

GitHub - google-deepmind/recurrentgemma: Open weights language model from Google DeepMind, based on Griffin. https://github.com/google-deepmind/recurrentgemma 38 comments

Linked pages

http://design 4 comments

Related searches:

Search whole site: site:jax.readthedocs.io

Search title: Pallas: a JAX kernel language — JAX documentation

See how to search.

Submit link to: