wang3702
Follow
Stars
Optax is a gradient processing and optimization library for JAX.
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Research and development (R&D) is crucial for the enhancement of industrial productivity, especially in the AI era, where the core aspects of R&D are mainly focused on data and models. We are commi…
A simple, performant and scalable Jax LLM!
ML Collections is a library of Python Collections designed for ML use cases.
Library for reading and processing ML training data.
Graph Attention Networks (https://arxiv.org/abs/1710.10903)
JAX - A curated list of resources https://github.com/google/jax
Flax is a neural network library for JAX that is designed for flexibility.
VARGPT: Unified Understanding and Generation in a Visual Autoregressive Multimodal Large Language Model
Code and models for the paper "The effectiveness of MAE pre-pretraining for billion-scale pretraining" https://arxiv.org/abs/2303.13496
Towards Scalable Pre-training of Visual Tokenizers for Generation
Awesome Unified Multimodal Models
High-resolution models for human tasks.
[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A curated list of awesome computational cryo-EM methods.
PyTorch implementation of JiT https://arxiv.org/abs/2511.13720
[NeurIPS 2025 D&B🔥] ImgEdit: A Unified Image Editing Dataset and Benchmark
The official GitHub repo for the survey paper "A Survey on Diffusion Language Models".
Official PyTorch implementation for "Large Language Diffusion Models"
Official inference repo for FLUX.2 models
MM-Vet: Evaluating Large Multimodal Models for Integrated Capabilities (ICML 2024)
Reformer, the efficient Transformer, in Pytorch
🟣 Computer Vision interview questions and answers to help you prepare for your next machine learning and data science interview in 2026.