A Jax implementation of word2vec's skip-gram model with negative sampling as described in Mikolov et al., 2013
-
Updated
Feb 22, 2021 - Python
A Jax implementation of word2vec's skip-gram model with negative sampling as described in Mikolov et al., 2013
A JAX implementation of DEformer for arbitrary conditioning.
Neural implicit digital elevation model
Kedro project of time-series-forecast of the multivariate weather using TSMixer.
Unofficial JAX/Flax implementation of Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutions.
Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
Molecular dynamics simulation using JAX library
A Simple Statistical Distribution Library in JAX
High-order Polynomial Projection Operators for JAX
A generalized implementation of Grad-CAM for Flax
minimal forward-mode automatic differentiation using python's abstract syntax tree
Research/master thesis project. A differentiable FEM model for vibrations of a Kirchoff-Love plate
Algorithms for inverse design
JAX implementation of Generalization and Exploration via Randomized Value Functions (Osband et al., 2016)
Fitting source models to Radio Interferometric visibilities using stochastic gradient descent.
Deep Reinforcement Learning with Jax
Common practices for distributed training using various backends
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."