https://github.com/moskomule/equinox_vision

Vision models with equinox
https://github.com/moskomule/equinox_vision

Last synced: 1 day ago
JSON representation

Vision models with equinox

• Host: GitHub
• URL: https://github.com/moskomule/equinox_vision
• Owner: moskomule
• License: mit
• Created: 2023-02-08T02:01:43.000Z (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2024-01-21T04:41:15.000Z (12 months ago)
• Last Synced: 2024-11-16T07:35:31.023Z (2 months ago)
• Language: Python
• Size: 63.5 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.txt

Awesome Lists containing this project

README

        
# equinox_vision

**This project is work in progress**

![pytest](https://github.com/moskomule/equinox_vision/workflows/pytest/badge.svg)

[![document](https://img.shields.io/static/v1?label=doc&message=equinox_vision&color=blue)](https://moskomule.github.io/equinox_vision)

`equinox_vision` is a library for small-size image classification problems in JAX with equinox.

## Installation

```commandline

pip install "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

pip install -U -e .

```

## Examples

See [examples](./examples).