Skip to content

@midjourney Midjourney

Change the repository type filter

All

    Repositories list

    11 repositories

    • docs

      Public
      (deprecated) Source for Midjourney's official wiki
      CSS
      •71•387•2•0•Updated Sep 22, 2024Sep 22, 2024

    • flash-attention-jax

      Public
      Implementation of Flash Attention in Jax
      Python
      •
      MIT License
      •23•7•0•0•Updated Jul 17, 2024Jul 17, 2024

    • nanobind

      Public
      nanobind: tiny and efficient C++/Python bindings
      C++
      •
      BSD 3-Clause "New" or "Revised" License
      •193•1•0•0•Updated Mar 15, 2024Mar 15, 2024

    • hf-transformers

      Public
      🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      •
      Apache License 2.0
      •27k•3•0•0•Updated Feb 14, 2024Feb 14, 2024

    • TransformerEngine

      Public
      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.
      Python
      •
      Apache License 2.0
      •310•2•0•0•Updated Jan 11, 2024Jan 11, 2024

    • transformers

      Public
      🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      •
      Apache License 2.0
      •27k•55•0•0•Updated 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
      •135•0•0•0•Updated May 1, 2023May 1, 2023

    • einops

      Public
      Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
      Python
      •
      MIT License
      •348•6•0•0•Updated Dec 29, 2022Dec 29, 2022

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.2k•3•0•0•Updated Dec 17, 2022Dec 17, 2022

    • jax

      Public
      Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
      Python
      •
      Apache License 2.0
      •2.8k•13•0•0•Updated Oct 12, 2022Oct 12, 2022

    • flax

      Public
      Flax is a neural network library for JAX that is designed for flexibility.
      Python
      •
      Apache License 2.0
      •633•9•0•0•Updated Sep 17, 2022Sep 17, 2022