ericjang/pt-jax external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

ericjang/pt-jax

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/ericjang/pt-jax)

Close

ericjang / pt-jaxView on GitHubMore

• External links
• Readme badge

Path Tracing in JAX

☆73Dec 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

• iree-org / iree-jax

  View on GitHub
  ☆55Aug 9, 2024Updated last year
• akshayka / hessian_trace_estimation

  View on GitHub
  Hessian trace estimation using PyTorch and Hutch++
  ☆20Oct 29, 2020Updated 5 years ago
• ericjang / nf-jax

  View on GitHub
  Normalizing Flows in Jax
  ☆107Aug 19, 2020Updated 5 years ago
• mattjj / ode-diff-notes

  View on GitHub
  ☆12Oct 8, 2020Updated 5 years ago
• j-towns / flipper

  View on GitHub
  Reversible programming in Agda
  ☆13Jun 22, 2023Updated 2 years ago
• yobibyte / pgn

  View on GitHub
  Graph Nets in pytorch
  ☆27Dec 8, 2022Updated 3 years ago
• mlysy / rodeo-legacy

  View on GitHub
  Probabilistic Solution of Differential Equations
  ☆13Jun 19, 2022Updated 3 years ago
• google-deepmind / trecvit

  View on GitHub
  ☆22Jan 10, 2026Updated last month
• Kajiyu / kanerva_machine

  View on GitHub
  The implementation of "The Kanerva Machine" with Pytorch and Pyro
  ☆12Jun 14, 2018Updated 7 years ago
• hardmaru / sema-demo

  View on GitHub
  sketch-rnn demo for seoul mediacity biennale 2018
  ☆13Sep 4, 2018Updated 7 years ago
• n2cholas / progan-flax

  View on GitHub
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12May 24, 2021Updated 4 years ago
• ArchRobison / SeismicDuck

  View on GitHub
  Reflection seismology game
  ☆13Sep 14, 2017Updated 8 years ago
• FLAIROx / JaxGL

  View on GitHub
  Simple JAX Graphics Library.
  ☆36Nov 3, 2024Updated last year
• BachiLi / redner

  View on GitHub
  Differentiable rendering without approximation.
  ☆1,434Aug 19, 2022Updated 3 years ago
• vaibhavsaxena11 / minerl_navigate

  View on GitHub
  MineRL Navigate Video Dataset
  ☆13Mar 24, 2021Updated 4 years ago
• jeremiecoullon / SGMCMCJax

  View on GitHub
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Oct 23, 2023Updated 2 years ago
• gdebie / stochastic-deep-networks

  View on GitHub
  Stochastic Deep Networks
  ☆17Jun 6, 2019Updated 6 years ago
• google-deepmind / dm_pix

  View on GitHub
  PIX is an image processing library in JAX, for JAX.
  ☆434Mar 6, 2025Updated 11 months ago
• j-towns / vdvae-jax

  View on GitHub
  Very deep VAEs in JAX/Flax
  ☆46Jun 16, 2021Updated 4 years ago
• sungyubkim / amortized_svgd

  View on GitHub
  A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN
  ☆18Dec 13, 2018Updated 7 years ago
• mmirman / MentisOculi

  View on GitHub
  A raytracer written in PyTorch (raynet?)
  ☆245Jun 5, 2020Updated 5 years ago
• Kaixhin / EC

  View on GitHub
  Episodic Control
  ☆22Sep 20, 2022Updated 3 years ago
• google / spectral-density

  View on GitHub
  Hessian spectral density estimation in TF and Jax
  ☆125Sep 6, 2020Updated 5 years ago
• dfm / extending-jax

  View on GitHub
  Extending JAX with custom C++ and CUDA code
  ☆403Aug 18, 2024Updated last year
• juliuskunze / cwvae-jax

  View on GitHub
  Clockwork VAEs in JAX/Flax
  ☆32Jul 16, 2021Updated 4 years ago
• mgallieriac / LASSO-MPC

  View on GitHub
  PhD Publications and Thesis on LASSO Model Predictive Control
  ☆20Jun 2, 2019Updated 6 years ago
• matthias-wright / flaxmodels

  View on GitHub
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆265Mar 21, 2025Updated 11 months ago
• arpastrana / jax_fdm

  View on GitHub
  Auto-differentiable and hardware-accelerated force density method
  ☆96Updated this week
• google-deepmind / jmp

  View on GitHub
  JMP is a Mixed Precision library for JAX.
  ☆211Jan 30, 2025Updated last year
• ArjunTeh / AdjointNonlinearRayTracing

  View on GitHub
  Differentiable ray tracing through inhomogeneous media
  ☆21Jun 12, 2022Updated 3 years ago
• n2cholas / shapecheck

  View on GitHub
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46May 4, 2021Updated 4 years ago
• ermongroup / dail

  View on GitHub
  The Official Implementation of Domain Adaptive Imitation Learning (DAIL)
  ☆24Oct 26, 2020Updated 5 years ago
• MilesCranmer / sympy2jax

  View on GitHub
  Turning SymPy expressions into JAX functions
  ☆45Mar 17, 2021Updated 4 years ago
• vaibhavsaxena11 / cwvae

  View on GitHub
  Clockwork Variational Autoencoder
  ☆50Jun 30, 2021Updated 4 years ago
• sail-sg / jrystal

  View on GitHub
  A JAX-based Differentiable Density Functional Theory Framework for Materials
  ☆45Feb 15, 2026Updated 2 weeks ago
• google-deepmind / einshape

  View on GitHub
  ☆109Jun 25, 2024Updated last year
• cgarciae / treex

  View on GitHub
  A Pytree Module system for Deep Learning in JAX
  ☆213Feb 26, 2023Updated 3 years ago
• google-deepmind / dm_construction

  View on GitHub
  ☆28Jan 11, 2021Updated 5 years ago
• fancompute / ising-compiler

  View on GitHub
  🍰 Compiling your code to an Ising Hamiltonian so you don't have to!
  ☆53Apr 30, 2020Updated 5 years ago