abarcel / unet-jaxLinks
Image segmentation with a U-Net-like architecture based on https://keras.io/examples/vision/oxford_pets_image_segmentation/
☆10Updated 3 years ago
Alternatives and similar repositories for unet-jax
Users that are interested in unet-jax are comparing it to the libraries listed below
Sorting:
- Geometric kernels on manifolds, meshes and graphs☆264Updated 3 weeks ago
- Use Jax functions in Pytorch☆248Updated 2 years ago
- ☆275Updated 2 years ago
- Newton and Quasi-Newton optimization with PyTorch☆358Updated 2 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆112Updated 5 months ago
- A library for programmatically generating equivariant layers through constraint solving☆271Updated 2 years ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆456Updated 9 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆30Updated 2 years ago
- Sparse nonlinear least squares in JAX☆227Updated 2 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆56Updated 2 weeks ago
- Rigid transforms + Lie groups for JAX☆289Updated 3 months ago
- Differentiable scientific computing library☆154Updated last year
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆627Updated 2 weeks ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆258Updated 2 years ago
- Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras☆28Updated 7 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆460Updated last week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆987Updated 4 months ago
- Exact OU processes with JAX☆50Updated 4 months ago
- A parallel ODE solver for PyTorch☆264Updated 10 months ago
- Run PyTorch in JAX. 🤝☆270Updated this week
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆58Updated 2 years ago
- Gaussian processes in JAX and Flax.☆525Updated last month
- Pytrees + dataclasses ❤️☆67Updated 3 months ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆85Updated last year
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆33Updated last year
- Algorithms for computations on random manifolds made easier☆92Updated last year
- Efficiently Composable Data Augmentation on the GPU with Jax☆39Updated 2 months ago
- Large-scale, multi-GPU capable, kernel solver☆190Updated 3 weeks ago
- Turn SymPy expressions into trainable JAX expressions.☆347Updated 3 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆130Updated 10 months ago