jax
Here are 947 public repositories matching this topic...
Common practices for distributed training using various backends
-
Updated
Sep 23, 2023 - Python
Code for various probabilistic deep learning models
-
Updated
Jun 28, 2023 - Jupyter Notebook
Conversion of LAST library from JAX to PyTorch
-
Updated
Dec 14, 2023 - Python
Python code for "Probabilistic Machine learning" book by Kevin Murphy
-
Updated
Feb 11, 2023 - Jupyter Notebook
Searching for galaxy satellites by their impact on lens potential in strong gravitational lensing setup
-
Updated
Sep 12, 2021 - Jupyter Notebook
This is the official repository of the paper "RoCourseNet: Distributionally Robust Training of a Prediction Aware Recourse Model".
-
Updated
Oct 22, 2023 - Jupyter Notebook
NXML is an eXtension for Machine Learning
-
Updated
Aug 17, 2022 - Python
implicit differentiation with jax
-
Updated
Nov 8, 2022 - Python
-
Updated
Aug 9, 2022 - Python
Denoising diffusion operators in JAX/FLAX
-
Updated
Apr 6, 2024 - 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
-
Updated
Mar 19, 2023 - Jupyter Notebook
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."