| --- |
| title: README |
| emoji: 🦀 |
| colorFrom: purple |
| colorTo: gray |
| sdk: static |
| pinned: false |
| --- |
| <p align="center"> |
| <image src="https://cdn-uploads.huggingface.co/production/uploads/6262be8108481d1d75edaad0/KeNFkl5dB4R2MMbH7IUhE.png" width="500"/> |
| </p> |
|
|
| ### JaxNN: Foundation Models in JAX/Flax |
|
|
| **Jax** **N**eural **N**etworks is an open-source library for foundation models in JAX and Flax. It provides a unified framework for loading, creating, and using pretrained models (e.g., ResNet, ViT). |
|
|
| - [HuggingFace models](https://huggingface.co/models?library=jaxnn&sort=trending) |
| - [Github repository](https://github.com/Xrenya/jaxnn) |
