jax-flash-attn2

Deep Learning for humans

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

? A ranked list of awesome machine learning Python libraries. Updated weekly.

Convert Machine Learning Code Between Frameworks

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Trax — Deep Learning with Clear Code and Speed

中文LLaMA-2 & Alpaca-2大模型二期项目 + 64K超长上下文模型 (Chinese LLaMA-2 & Alpaca-2 LLMs with 64K long context models)

Official release of InternLM series (InternLM, InternLM2, InternLM2.5, InternLM3).

Python code for "Probabilistic Machine learning" book by Kevin Murphy

Flax is a neural network library for JAX that is designed for flexibility.