May 16, 2024, 3:23 a.m. | /u/RocketLL

Machine Learning www.reddit.com

I created jaxsplat which provides CUDA-accelerated 3D Gaussian Splatting for JAX.
The original INRIA code and gsplat's implementation contain dynamically shaped arrays unsuitable for usage with JAX.
Instead, I modified gsplat's CUDA implementation to expose custom XLA CUDA calls while not leaking any dynamic shapes into JAX-side code.

Take a look if you're interested in exploring 3D Gaussian Splatting with JAX:

GitHub: https://github.com/yklcs/jaxsplat

Docs: https://jaxsplat.readthedocs.io

arrays code cuda dynamic implementation jax look machinelearning usage while xla

Doctoral Researcher (m/f/div) in Automated Processing of Bioimages

@ Leibniz Institute for Natural Product Research and Infection Biology (Leibniz-HKI) | Jena

Seeking Developers and Engineers for AI T-Shirt Generator Project

@ Chevon Hicks | Remote

Software Engineer for AI Training Data (School Specific)

@ G2i Inc | Remote

Software Engineer for AI Training Data (Python)

@ G2i Inc | Remote

Software Engineer for AI Training Data (Tier 2)

@ G2i Inc | Remote

Principal Research Engineer - Materials

@ GKN Aerospace | Westlake, TX, US