whatisgithub

What is omnix?

yanqinjiang/omnix — explained in plain English

Analysis updated 2026-05-18

87PythonAudience · researcherComplexity · 5/5Setup · hard

In one sentence

Official research code for a 4D scene reconstruction paper that predicts moving 3D scenes from video in a single neural network pass.

Mindmap

mindmap
  root((repo))
    What it does
      Reconstructs 4D scenes
      Any view any time
      Feed forward trajectory fields
    Tech stack
      Python
      PyTorch
      FlashAttention-3
      CUDA
    Use cases
      Reproduce paper results
      Run inference on sample scenes
      Train on custom video data
    Audience
      Computer vision researchers
      Graduate students
      AI research labs
    Requirements
      Nvidia Hopper or Blackwell GPU
      Pretrained checkpoint
      Custom compiled operators
    Origin
      CASIA
      Tencent Hunyuan
      HKUST

Code map

Detail Auto

An interactive map of this repo's files and how they connect — its source is parsed live in your browser. Click Visualize to build it.

filefunction / class

What do people build with it?

USE CASE 1

Reproduce the paper's 4D scene reconstruction results using the provided pretrained checkpoint.

USE CASE 2

Run inference on your own images to reconstruct a moving 3D scene.

USE CASE 3

Train the model on custom video datasets after running the preprocessing pipeline.

USE CASE 4

Study a feed forward approach to 4D reconstruction as a basis for further research.

What is it built with?

PythonPyTorchCUDAFlashAttention-3

How does it compare?

yanqinjiang/omnixamazon-science/cyber-zeroluyi14-bits/jig
Stars878787
LanguagePythonPythonPython
Setup difficultyhardhardmoderate
Complexity5/54/54/5
Audienceresearcherresearcherdeveloper

Figures from each repo's GitHub metadata at analysis time.

How do you get it running?

Difficulty · hard Time to first run · 1day+

Requires an Nvidia Hopper or Blackwell GPU and compiling FlashAttention-3 and Deformable DETR operators from source.

No license file is mentioned in the README, so permitted uses are unclear.

So what is it?

OmniX is the official code release for an academic research paper about rebuilding moving, three dimensional scenes from video, a task researchers call 4D reconstruction because it captures both space and time. The project comes from researchers at CASIA, Tencent Hunyuan, and HKUST, and it was accepted at a major computer vision conference. The core idea is to take footage from any set of camera views and any point in time, and predict how the scene looks and moves in 3D, using a single trained neural network rather than needing to run a separate optimization process for each new scene. The team frames this as learning trajectory fields, a way of describing how points in the scene move over time, and their model produces this in a single forward pass rather than a slow iterative one. This is research code, not a consumer product. Using it requires a capable Nvidia GPU, specifically ones from Nvidia's newer Hopper or Blackwell generations, since the code depends on a specialized attention technique called FlashAttention-3 that needs that hardware. Setup involves creating a Python environment, installing a matching version of PyTorch for your CUDA version, and compiling two additional pieces of code from source, including a specialized attention library and detection operators. Once set up, you can either run inference using a pretrained model checkpoint the authors provide to reconstruct a sample scene from images, or train the model yourself on your own data after preprocessing it with tools borrowed from a related project. The authors note that the provided training settings are a starting point and will likely need adjustment for different hardware or datasets. The project builds on and credits several other open source research efforts, and the authors ask that anyone who finds it useful cite their paper.

Copy-paste prompts

Prompt 1
Explain what any view and any time 4D reconstruction means in plain terms.
Prompt 2
Walk me through installing OmniX and its FlashAttention-3 dependency on a compatible GPU.
Prompt 3
Help me run inference with the pretrained OmniX checkpoint on my own test images.
Prompt 4
Explain what a feed forward trajectory field is and how it differs from per scene optimization.

Frequently asked questions

What is omnix?

Official research code for a 4D scene reconstruction paper that predicts moving 3D scenes from video in a single neural network pass.

What language is omnix written in?

Mainly Python. The stack also includes Python, PyTorch, CUDA.

What license does omnix use?

No license file is mentioned in the README, so permitted uses are unclear.

How hard is omnix to set up?

Setup difficulty is rated hard, with roughly 1day+ to a first successful run.

Who is omnix for?

Mainly researcher.

Open on GitHub → Ask about another repo

This repo across BitVibe Labs

Verify against the repo before relying on details.