15 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Task-based datasets, preprocessing, and evaluation for sequence models. | 347K | |
| Task-based datasets, preprocessing, and evaluation for sequence models. | 216K | |
| A JAX-native High Performance Eval Metrics Library | 35K | |
| Functionalities such as a layers for building neural networks in Jax. | 31K | |
| Pax is a Jax-based machine learning framework for training large scale models. P... | 29K | |
| Modular, scalable library to train ML models | 24K | |
| A versatile and easy to understand Diffusion library | 3K | |
| REAX — Scalable, flexible training for JAX, inspired by the simplicity of PyTorc... | 2K | |
| Integration of scRNA-seq and spatial transcpriptomics data | 1K | |
| A flexible and performant framework for training machine learning potentials. | 973 | |
| Transformer based embeddings for Wasserstein Distances | 576 | |
| LLM post-training library | 319 | |
| Strong lensing package using jax | 130 | |
| MT3 Music Transcription Library | 107 | |
| Lingvo JAX libraries. | 84 |
