Similar repositories to lucidrains/flash-attention-jax:

• lucidrains/flash-attention-jax github similar
• srush/GPU-Puzzles github similar
• lucidrains/flash-cosine-sim-attention github similar
• lucidrains/parti-pytorch github similar
• yandex/YaLM-100B github similar
• nnaisense/evotorch github similar
• openai/Video-Pre-Training github similar
• meituan/YOLOv6 github similar
• lab-ml/annotated_deep_learning_paper_implementations github similar
• thomasneff/AdaNeRF github similar
• moyix/fauxpilot github similar
• patil-suraj/vit-vqgan github similar
• hanxiao/bert-as-service github similar
• uzh-rpg/rpg_ultimate_slam_open github similar
• toeverything/AFFiNE github similar
• michaelgutmann/ml-pen-and-paper-exercises github similar
• salesforce/CodeRL github similar
• lucidrains/x-unet github similar
• shyamsn97/hyper-nn github similar
• AssemblyAI-Examples/MinImagen github similar
• pesser/stable-diffusion github similar
• TimDettmers/bitsandbytes github similar
• lucidrains/transframer-pytorch github similar
• microsoft/GODEL github similar
• danistefanovic/build-your-own-x github similar
• lucidrains/gated-state-spaces-pytorch github similar
• xplip/pixel github similar
• omerbt/Text2LIVE github similar
• rinongal/textual_inversion github similar
• patr-schm/TinyAD github similar
• mitsuba-renderer/mitsuba3 github similar
• r00tman/NeRF-OSR github similar
• lucidrains/tranception-pytorch github similar
• lucidrains/retrieval-augmented-ddpm github similar
• lucidrains/flexible-diffusion-modeling-videos-pytorch github similar
• mortacious/cupy-knn github similar
• lucidrains/metaformer-gpt github similar
• ylsung/Ladder-Side-Tuning github similar
• openai/human-eval-infilling github similar
• rshaojimmy/SeqDeepFake github similar