nerf.jax

A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.

Stars
11
View Code on GitHub Visit Website View on X
Ecosystems: Python

Statistics for this project are still being loaded, please check back later.

Related Projects
vqgan-jax

JAX implementation of VQGAN

28 Jun 2021 90
nerfies

This is the code for Deformable Neural Radiance Fields, a.k.a. Nerfies.

21 Jan 2021 1,636
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
stable-diffusion-jax

12 Aug 2022 85
tensorf-jax

Unofficial implementation of Tensorial Radiance Fields (Chen & Xu ‘22)

17 Apr 2022 39
serket

The ✨Magical✨ JAX ML Library.

20 Aug 2022 16
sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.sit...

03 Oct 2023 311
jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...

23 Jun 2022 1,131
jax_transformer

Autoregressive transformer in JAX from scratch

31 Aug 2021 14
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
sympy2jax

Turning SymPy expressions into JAX functions

26 Feb 2021 41
nerf-pytorch

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

05 Apr 2020 5,409
diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs....

25 Jul 2021 1,390