jax
Here are 934 public repositories matching this topic...
A Jax implementation of word2vec's skip-gram model with negative sampling as described in Mikolov et al., 2013
-
Updated
Feb 22, 2021 - Python
Toy implementations of popular deep learning optimizers from scratch in JAX
-
Updated
Jun 20, 2021 - Jupyter Notebook
This is an implementation of linear regression using JAX and a functional programming approach. You can find a related blog post at the following link: https://medium.com/@sahinadirhan/simple-linear-regression-using-jax-5ef2eefb8cf4
-
Updated
Jul 17, 2021 - Jupyter Notebook
A JAX implementation of DEformer for arbitrary conditioning.
-
Updated
May 29, 2022 - Python
Neural implicit digital elevation model
-
Updated
May 25, 2022 - Python
-
Updated
Mar 19, 2023 - Jupyter Notebook
Research/master thesis project. A differentiable FEM model for vibrations of a Kirchoff-Love plate
-
Updated
Apr 11, 2022 - Python
Kedro project of time-series-forecast of the multivariate weather using TSMixer.
-
Updated
Oct 10, 2023 - Python
Unofficial JAX/Flax implementation of Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutions.
-
Updated
Dec 15, 2023 - Python
Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
-
Updated
Jul 18, 2023 - Python
Simple VAE implementation in tensorflow, jax and pytorch where both the encoder and decoder model use gaussian distributions.
-
Updated
Mar 4, 2024 - Jupyter Notebook
Machine learning samples
-
Updated
Mar 21, 2024
Molecular dynamics simulation using JAX library
-
Updated
Mar 12, 2024 - Python
A Simple Statistical Distribution Library in JAX
-
Updated
Mar 30, 2024 - Python
High-order Polynomial Projection Operators for JAX
-
Updated
Feb 22, 2023 - Python
A JAX implementation to unify machine learning paradigms into a single design pattern
-
Updated
May 18, 2024 - Python
Improve this page
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."