Skip to content
Change the repository type filter

All

    Repositories list

    • foo23452

      Public
      Hardware Bot public-facing content
      0000Updated Mar 13, 2026Mar 13, 2026
    • docs

      Public
      (deprecated) Source for Midjourney's official wiki
      CSS
      6938820Updated Mar 9, 2026Mar 9, 2026
    • Quantized Attention achieves speedup of 2-5x and 3-11x compared to FlashAttention and xformers, without lossing end-to-end metrics across language, image, and v…
      Cuda
      Apache License 2.0
      375200Updated Jul 8, 2025Jul 8, 2025
    • xDiT

      Public archive
      xDiT: A Scalable Inference Engine for Diffusion Transformers (DiTs) with Massive Parallelism
      Python
      Apache License 2.0
      314100Updated Jun 18, 2025Jun 18, 2025
    • USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference
      Python
      Apache License 2.0
      79200Updated Jun 5, 2025Jun 5, 2025
    • Implementation of Flash Attention in Jax
      Python
      MIT License
      25900Updated Jul 17, 2024Jul 17, 2024
    • nanobind

      Public
      nanobind: tiny and efficient C++/Python bindings
      C++
      BSD 3-Clause "New" or "Revised" License
      286200Updated Mar 15, 2024Mar 15, 2024
    • 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      Apache License 2.0
      33k600Updated Feb 14, 2024Feb 14, 2024
    • 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      Apache License 2.0
      33k5600Updated Sep 1, 2023Sep 1, 2023
    • jaxtorch

      Public
      Python
      MIT License
      9100Updated Jun 7, 2023Jun 7, 2023
    • equinox

      Public archive
      Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/
      Python
      Apache License 2.0
      181100Updated May 1, 2023May 1, 2023
    • einops

      Public
      Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
      Python
      MIT License
      396700Updated Dec 29, 2022Dec 29, 2022
    • Fast and memory-efficient exact attention
      Python
      BSD 3-Clause "New" or "Revised" License
      2.5k400Updated Dec 17, 2022Dec 17, 2022
    • jax

      Public archive
      Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
      Python
      Apache License 2.0
      3.5k1500Updated Oct 12, 2022Oct 12, 2022
    • flax

      Public archive
      Flax is a neural network library for JAX that is designed for flexibility.
      Python
      Apache License 2.0
      7931000Updated Sep 17, 2022Sep 17, 2022