| Name | Version | Summary | date |
| juxai-s2 |
3.0.0 |
JUX is a jax-accelerated engine for Lux-2022. |
2023-09-27 03:35:26 |
| jax_chmc |
0.0.5 |
An implementation of a family of MCMC methods on implicitly defined manifolds |
2023-09-20 16:26:59 |
| jimGW |
0.1.1 |
Gravitatioanl wave data analysis tool in Jax |
2023-09-18 20:10:19 |
| hf-image-uploader |
0.0.2 |
State-of-the-art diffusion in PyTorch and JAX. |
2023-09-05 13:19:41 |
| holab |
0.0.3 |
Acoustic lens design on GPUs with Automatic Differentiation. |
2023-09-04 16:17:10 |
| nm-transformers |
1.5.1.42301 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2023-08-23 01:27:39 |
| tensorshare |
0.1.1 |
🤝 Trade any tensors over the network |
2023-08-20 11:06:38 |
| veros-extra-setups |
1.5.0 |
Extra setups for Veros, the versatile ocean simulator. |
2023-08-18 23:05:53 |
| serket |
0.1.0 |
Finite difference tools in JAX |
2023-07-27 06:38:35 |
| nebula-loss |
0.4.1 |
1 Loss Function to rule them all! |
2023-07-25 15:34:06 |
| connex |
0.3.3 |
Fine-grained, dynamic control of neural network topology in JAX. |
2023-07-23 05:22:47 |
| ezjaxtyping |
0.2.20 |
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. |
2023-07-20 18:27:43 |
| flash-attention-jax |
0.3.1 |
Flash Attention - in Jax |
2023-07-18 02:45:18 |
| arrgh |
0.0.1 |
A small python utility to pretty-print a table summarizing arrays & scalars from numpy, pytorch, etc. |
2023-06-15 23:32:42 |
| kernex |
0.2.0 |
Stencil computations in JAX. |
2023-06-09 21:12:04 |
| FiniteDiffX |
0.0.5 |
Finite difference tools in JAX. |
2023-06-08 12:45:53 |
| pcax |
0.1.0 |
Minimal Principal Component Analysis (PCA) implementation using JAX. |
2023-05-29 12:20:21 |
| jit-env |
0.1.4 |
A Jax interface for Reinforcement Learning environments. |
2023-05-12 13:32:07 |
| jaxdlm |
0.0.2 |
A Jax based python package for dynamic linear models |
2023-05-11 19:28:09 |
| teneva-jax |
0.1.1 |
Compact implementation of basic operations in the tensor-train (TT) format with jax framework for approximation, optimization and sampling with multidimensional arrays and multivariate functions |
2023-04-24 20:58:22 |