Midjourney
Popular repositories
-
transformers
transformers PublicForked from huggingface/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
-
jax
jax PublicForked from google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
flax
flax PublicForked 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
- collective_bots Public archive Forked from Rapptz/discord.py
discord bots for collective intelligence
- hf-transformers Public Forked from GallagherCommaJack/hf-transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
- TransformerEngine Public Forked from NVIDIA/TransformerEngine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
- 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-jax Public Forked from lucidrains/flash-attention-jax
Implementation of Flash Attention in Jax
- flash-attention Public Forked from Dao-AILab/flash-attention
Fast and memory-efficient exact attention