Midjourney
Popular repositories Loading
-
transformers
transformers PublicForked from huggingface/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
-
jax
jax Public archiveForked from jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
flax
flax Public archiveForked from IvyZX/flax
Flax is a neural network library for JAX that is designed for flexibility.
-
flash-attention-jax
flash-attention-jax PublicForked from lucidrains/flash-attention-jax
Implementation of Flash Attention in Jax
-
einops
einops PublicForked from arogozhnikov/einops
Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
Repositories
- flash-attention-jax Public Forked from lucidrains/flash-attention-jax
Implementation of Flash Attention in Jax
- hf-transformers Public Forked from GallagherCommaJack/hf-transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
- transformers Public Forked from huggingface/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
- equinox Public archive Forked from patrick-kidger/equinox
Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/
- flash-attention Public Forked from Dao-AILab/flash-attention
Fast and memory-efficient exact attention