jax2torch

Use Jax functions in Pytorch

MIT License

Downloads
345
Stars
224
Committers
1
View Code on GitHub
Ecosystems: Python

Commit Statistics

Past Year

All Time

Total Commits
14
Total Committers
1
Avg. Commits Per Committer
14.0
Bot Commits
0

Issue Statistics

Past Year

All Time

Total Pull Requests
0
1
Merged Pull Requests
0
1
Total Issues
1
3
Time to Close Issues
about 1 month
18 days
Package Rankings
Top 14.67% on Pypi.org
Badges
Extracted from project README
Open In Colab
Related Projects
jaxtyping

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

23 Jun 2022 1,131
tensor-sensor

The goal of this library is to generate more helpful exception messages for matrix algebra expres...

28 Aug 2020 771
annotated-s4

Implementation of https://srush.github.io/annotated-s4

08 Dec 2021 450
fidjax

Frechet Inception Distance in JAX

15 Jul 2024 1
jpu

JAX + Units

24 May 2022 35
equinox

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

29 Jul 2021 1,846
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
parallax

19 May 2020 157
lineax

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

09 May 2023 345
sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314
tensorf-jax

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

17 Apr 2022 39
diffrax

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

25 Jul 2021 1,390
sympy2jax

Turning SymPy expressions into JAX functions

26 Feb 2021 41
jax-gpu-cluster-demo

Little demo of how to instantiate a cluster of GPUs with JAX

21 Jan 2022 6
eagerpy

PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code

23 Oct 2019 695