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: almost 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. | 238 |
 erfanzar/easydel | A flexible framework for training and serving machine learning models with JAX/Flax | 206 |
 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 | A Python-based framework for training Conditional Random Field Recurrent Neural Networks (CRF-RNN) for semantic image segmentation | 199 |
 vpgtrans/vpgtrans | Transfers visual prompt generators across large language models to reduce training costs and enable customization of multimodal LLMs | 269 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 461 |
 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 | 790 |
 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,300 |
 minimaxir/automl-gs | Automates machine learning model creation and optimization for complex datasets | 1,853 |
 stevedoyle2/pynastran | An interface library to Nastran file formats, allowing users to read and manipulate model geometry and results data. | 396 |
 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,782 |