AIbase
Home
AI Tutorial

jimm

Public

JAX Image Modeling of Models contains Computer Vision/Vision Language Model implementations in native flax nnx with proper sharding annotations (allowing you to do easy fully sharded data parallel training with native flax nnx/jax)

computer-visionflaxjaxmachine-learning
Creat：2025-06-05T15:11:31
Update：2025-06-18T07:54:46
https://pythoncrazy.github.io/jimm/
5
Stars
0
Stars Increase

Related projects

Computer Science

Hot
awesome-list

? Path to a free self-taught education in Computer Science!

198852
8个月前
+153today

Transformers

Hot
bert

? Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

153615
2年前
+136today

Opencv

c-plus-plus

Open Source Computer Vision Library

85202
7年前
+42today

D2l Zh

Hot
book

《动手学深度学习》：面向中文读者、能运行、可讨论。中英文版被70多个国家的500多所大学用于教学。

74277
8个月前
+60today

Cs Video Courses

Hot
algorithms

List of Computer Science courses with video lectures.

70343
8个月前
+91today

Awesome Scalability

Hot
architecture

The Patterns of Scalable, Reliable, and Performant Large-Scale Systems

67088
8个月前
+52today

Awesome Courses

Hot
awesome

:books: List of awesome university courses for learning Computer Science!

64946
8个月前
+70today

Keras

data-science

Deep Learning for humans

63624
8个月前
-3today

AI For Beginners

ai

12 Weeks, 24 Lessons, AI for All!

44127
8个月前
+44today

Pytorch Image Models

augmix

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

35969
8个月前
+24today