Skip to content
Change the repository type filter

All

    Repositories list

    • docs

      Public
      (deprecated) Source for Midjourney's official wiki
      CSS
      7038820Updated Nov 12, 2024Nov 12, 2024
    • Implementation of Flash Attention in Jax
      Python
      MIT License
      23700Updated Jul 17, 2024Jul 17, 2024
    • nanobind

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

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

      Public
      Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
      Python
      MIT License
      352600Updated Dec 29, 2022Dec 29, 2022
    • Fast and memory-efficient exact attention
      Python
      BSD 3-Clause "New" or "Revised" License
      1.3k300Updated 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
      2.8k1300Updated 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
      648900Updated Sep 17, 2022Sep 17, 2022