Skip to content
Change the repository type filter

All

    Repositories list

    • docs

      Public
      (deprecated) Source for Midjourney's official wiki
      CSS
      6838620Updated Jun 29, 2026Jun 29, 2026
    • Get all of the files changed/modified in a pull request or push's commits.
      TypeScript
      MIT License
      146000Updated Jun 4, 2026Jun 4, 2026
    • foo23452

      Public
      Hardware Bot public-facing content
      0000Updated Mar 13, 2026Mar 13, 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
      439100Updated 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
      325100Updated 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
      80100Updated 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
      314200Updated Mar 15, 2024Mar 15, 2024
    • 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      Apache License 2.0
      34k500Updated Feb 14, 2024Feb 14, 2024
    • 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      Apache License 2.0
      34k5600Updated 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
      203100Updated May 1, 2023May 1, 2023
    • einops

      Public
      Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
      Python
      MIT License
      400700Updated Dec 29, 2022Dec 29, 2022
    • Fast and memory-efficient exact attention
      Python
      BSD 3-Clause "New" or "Revised" License
      2.9k400Updated 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.7k1400Updated 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
      8201000Updated Sep 17, 2022Sep 17, 2022
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.