palm-jax (pypi): PaLM: Scaling Language Modeling with Pathways - Jax
Keywords: artificial intelligence|deep learning|transformers|attention mechanism|artificial-general-intelligence|attention-mechanism|deep-learning|jax