brianfitzgerald/jax-mmdit brianfitzgerald/jax-mmdit

  • Stars
    star
    15
  • Rank 1,364,093 (Top 27 %)
  • Language
    Python
  • Created 4 months ago
  • Updated 3 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
brianfitzgerald/jax-mmdit
github

brianfitzgerald

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Implementation of Diffusion Transformers and Rectified Flow in Jax

More Repositories

1

style_aligned_comfy

Python
261
star
2

Comfy-SVDTools

Python
41
star
3

superprompt

Python
14
star
4

smolmodels

Collection of experiments related to small language models, mostly seq2seq models
Python
8
star
5

snap-templates

Template syntax for easy GraphQL service integration
JavaScript
2
star
6

burritobot

keeping the burrito economy afloat
Go
2
star
7

panopticon

screenshot testing for the web
TypeScript
1
star
8

spotify-reducer

Go
1
star
9

depth_diffusion

Jupyter Notebook
1
star
10

cdk-lambda-deploy-poc

TypeScript
1
star
11

time-series-prediction

A number of implementations of various time series prediction models and datasets
Python
1
star