| Name | Version | Summary | date |
| pcax |
0.1.0 |
Minimal Principal Component Analysis (PCA) implementation using JAX. |
2023-05-29 12:20:21 |
| rax |
0.3.0 |
Learning-to-Rank using JAX. |
2023-05-17 19:24:49 |
| 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 |
| sequel-core |
0.0.2 |
A Continual Learning Framework for both Jax and PyTorch. |
2023-04-18 15:50:11 |
| transformers-machinify |
4.27.0.dev0 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2023-03-25 00:41:42 |
| flaim |
0.0.5 |
Flax Image Models |
2023-01-27 17:42:17 |
| jaxGW |
0.0.4.2 |
Gravitatioanl wave data analysis tool in Jax |
2023-01-26 15:20:02 |
| funsor |
0.4.5 |
A tensor-like library for functions and distributions |
2023-01-23 08:31:02 |
| jaxkern |
0.0.5 |
Kernels in Jax. |
2023-01-19 18:47:03 |
| gpjax-nightly |
0.5.8 |
Didactic Gaussian processes in Jax. |
2023-01-10 00:06:44 |
| jaxutils |
0.0.8 |
Utility functions for JaxGaussianProcesses |
2023-01-08 22:22:59 |
| juxai2022 |
1.0 |
JUX is a jax-accelerated engine for Lux-2022. |
2022-12-26 08:33:26 |
| jax-rex |
0.0.11 |
Jax-based Recourse Explanation Library |
2022-12-24 21:55:36 |
| JaxKern |
0.0.3 |
Kernels in Jax. |
2022-12-18 21:53:42 |
| GPJax |
0.5.5 |
Didactic Gaussian processes in Jax. |
2022-12-18 19:20:41 |
| JaxUtils |
0.0.3 |
Utility functions for JaxGaussianProcesses |
2022-12-16 14:24:23 |
| hamux |
0.1.1 |
A Deep Learning framework built around ENERGY |
2022-12-02 06:29:19 |
| jraph |
0.0.6.dev0 |
Jraph: A library for Graph Neural Networks in Jax |
2022-08-12 15:24:47 |