flaxvision
Model porters
A collection of neural network models adapted for JAX and Flax frameworks.
A selection of neural network models ported from torchvision for JAX & Flax.
44 stars
4 watching
2 forks
Language: Python
last commit: about 4 years ago
Linked from 1 awesome list
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 240 |
 erfanzar/easydel | A framework for training and serving machine learning models using JAX, with a focus on flexibility and customization. | 212 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 wmvanvliet/pytorch_hmax | An implementation of the HMAX model for vision processing in PyTorch | 57 |
 martinkersner/train-crf-rnn | Trains a CRF-RNN model for semantic image segmentation using the PASCAL VOC dataset. | 199 |
 vpgtrans/vpgtrans | Transfers visual prompt generators across large language models to reduce training costs and enable customization of multimodal LLMs | 270 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 467 |
 scicloj/scicloj.ml.clj-djl | Provides pre-trained machine learning models for natural language processing tasks using Clojure and the clj-djl framework. | 0 |
 molcik/python-neuron | A Python library for implementing and training various neural network architectures | 40 |
 mosecorg/mosec | A high-performance ML model serving framework | 802 |
 royson/fedl2p | This project enables personalized learning models by collaborating on learning the best strategy for each client | 19 |
 lxtgh/omg-seg | Develops an end-to-end model for multiple visual perception and reasoning tasks using a single encoder, decoder, and large language model. | 1,336 |
 minimaxir/automl-gs | Automates machine learning model creation and optimization for complex datasets | 1,857 |
 stevedoyle2/pynastran | An interface library to Nastran file formats, allowing users to read and manipulate model geometry and results data. | 395 |
 lyhue1991/torchkeras | A PyTorch-based model training framework designed to simplify and streamline training workflows by providing a unified interface for various loss functions, optimizers, and validation metrics. | 1,822 |