| Name | Version | Summary | date |
| tsnex |
0.0.1 |
Minimal t-distributed stochastic neighbor embedding (t-SNE) implementation in JAX. |
2024-03-12 17:34:37 |
| jax-chmc |
0.0.7 |
An implementation of a family of MCMC methods on implicitly defined manifolds |
2024-03-09 09:10:42 |
| cfrx |
0.0.2 |
Counterfactual Regret Minimization in Jax |
2024-03-05 14:45:34 |
| nm-transformers-nightly |
1.7.0.20240304 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-03-05 13:39:02 |
| bosonic-jax |
0.1.0.1 |
Bosonic JAX |
2024-03-04 10:57:02 |
| chromax |
0.0.4a0 |
Breeding simulator based on JAX |
2024-03-03 19:58:23 |
| harmonix |
0.0.1 |
Analytic interferometry of stellar surfaces using spherical harmonics in Jax |
2024-03-02 14:25:34 |
| google-fax |
0.0.2 |
FAX is an implementation of federated operations using Jax primitives. |
2024-02-29 21:44:46 |
| ref-gsd |
0.2.3 |
Reference implementation of generalised score distribution in python |
2024-02-26 07:22:58 |
| bayex |
0.2.0 |
Minimal Bayesian Optimization Implementation with Gaussian Processes written in JAX. |
2024-02-25 12:48:46 |
| drpangloss |
0.0.1 |
Python package to fit interferometric data, accelerated using Jax. |
2024-02-23 05:43:15 |
| arraylias |
0.1.1 |
A Python package for aliased function dispatching to multiple array libraries |
2024-02-20 19:00:09 |
| jaxbook |
0.0.1 |
Iterative algorithms for numerical linear algebra with JAX |
2024-02-19 06:39:00 |
| sparsejac |
0.1.3 |
Efficient forward- and reverse-mode sparse Jacobians using Jax. |
2024-02-18 17:07:45 |
| dm-reverb-nightly |
0.15.0.dev20240213 |
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research. |
2024-02-13 11:07:47 |
| jaxwt |
0.1.1 |
Differentiable and gpu enabled fast wavelet transforms in JAX |
2024-02-13 10:41:18 |
| ghq |
0.0.3 |
Gauss-Hermite quadrature in JAX |
2024-02-10 16:17:43 |
| flashbax |
0.1.0 |
Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation. |
2024-02-06 10:33:04 |
| jaxampler |
0.0.7 |
A JAX-based statistical sampling toolkit |
2024-01-27 20:33:03 |
| mlmarketing |
0.0.1 |
Machine Learning for Marketing |
2024-01-21 14:20:44 |