| Name | Version | Summary | date |
| easydel |
0.0.80 |
An open-source library to make training faster and more optimized in JAX |
2024-12-04 15:22:10 |
| stablehlo-coreml-experimental |
0.0.11 |
Convert StableHLO models into Apple Core ML format |
2024-11-20 18:25:18 |
| fjformer |
0.0.91 |
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer in JAX |
2024-11-14 10:11:07 |
| jax-flash-attn2 |
0.0.1 |
Flash Attention Implementation with Multiple Backend Support and Sharding This module provides a flexible implementation of Flash Attention with support for different backends (GPU, TPU, CPU) and platforms (Triton, Pallas, JAX). |
2024-10-23 22:37:19 |
| FJFormer |
0.0.66 |
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal of custom Jax Flax Functions and Utils that elevate your AI endeavors to new heights! |
2024-06-12 13:32:40 |
| EasyDeL |
0.0.67 |
An open-source library to make training faster and more optimized in Jax/Flax |
2024-06-02 17:14:40 |