---
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)