Path Tracing in JAX
☆75Dec 15, 2019Updated 6 years ago
Alternatives and similar repositories for pt-jax
Users that are interested in pt-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Probabilistic Solution of Differential Equations☆13Jun 19, 2022Updated 3 years ago
- Normalizing Flows in Jax☆109Aug 19, 2020Updated 5 years ago
- ☆12Oct 8, 2020Updated 5 years ago
- MineRL Navigate Video Dataset☆13Mar 24, 2021Updated 5 years ago
- ☆56Aug 9, 2024Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆81Nov 14, 2023Updated 2 years ago
- Very deep VAEs in JAX/Flax☆47Jun 16, 2021Updated 4 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆27Apr 22, 2021Updated 5 years ago
- Differentiable rendering without approximation.☆1,439Aug 19, 2022Updated 3 years ago
- Mathematical operations for JAX pytrees☆210Dec 5, 2024Updated last year
- A study of performance of optimal transport.☆10Jul 4, 2020Updated 5 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Jun 14, 2018Updated 7 years ago
- Simple JAX Graphics Library.☆37Nov 3, 2024Updated last year
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- A general-purpose lightweight sandbox for safely executing user programs☆16Jun 24, 2020Updated 5 years ago
- sketch-rnn demo for seoul mediacity biennale 2018☆13Sep 4, 2018Updated 7 years ago
- I/O operations for OME-TIFF files in Julia☆25Nov 1, 2024Updated last year
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Sep 28, 2022Updated 3 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 5 years ago
- ☆25Jan 10, 2026Updated 4 months ago
- Graph Nets in pytorch☆28Dec 8, 2022Updated 3 years ago
- Steklov Spectral Geometry for Extrinsic Shape Analysis☆12Aug 9, 2024Updated last year
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- A raytracer written in PyTorch (raynet?)☆245Jun 5, 2020Updated 5 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Feb 3, 2020Updated 6 years ago
- ☆28Jan 12, 2022Updated 4 years ago
- A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN☆18Dec 13, 2018Updated 7 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- Supporting material for Princeton ORF522☆14Aug 27, 2025Updated 9 months ago
- A self-hosted remote communication system designed for math discussions.☆41Dec 21, 2020Updated 5 years ago
- Episodic Control☆22Sep 20, 2022Updated 3 years ago
- Concise deep learning for JAX☆183Oct 21, 2020Updated 5 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- PIX is an image processing library in JAX, for JAX.☆438Mar 6, 2025Updated last year
- JMP is a Mixed Precision library for JAX.☆213Jan 30, 2025Updated last year
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆265Mar 21, 2025Updated last year
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- Sparse nonlinear least squares in JAX☆287Feb 26, 2026Updated 3 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆92Oct 19, 2022Updated 3 years ago