Pinned Loading

1. EasyDeL EasyDeL Public

   Accelerate, Optimize performance with streamlined training and serving options with JAX.

   Python 345 47

2. jax-flash-attn2 jax-flash-attn2 Public

   A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/Pallas/JAX).

   Python 34 1

3. eformer eformer Public

   (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX

   Python 29 7

4. jax-ml/jax jax-ml/jax Public

   Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

   Python 35.2k 3.5k

5. OST-OpenSourceTransformers OST-OpenSourceTransformers Public

   OST Collection: An AI-powered suite of models that predict the next word matches with remarkable accuracy (Text Generative Models). OST Collection is based on a novel approach to work as a full and…

   Jupyter Notebook 16 1

6. Calute Calute Public

   Agents for intelligence and coordination

   Python 20