Name | Version | Summary | date |
autobnn |
0.0.2.dev0 |
Package for training Gaussian process-like Bayesian Neural Networks with composite structure. |
2024-03-28 00:19:39 |
tf2jax |
0.3.6 |
TF2JAX: Convert TensorFlow to JAX |
2024-03-26 10:38:21 |
souJpg-diffusers |
0.27.2 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-03-25 00:20:06 |
hydrax |
0.2.2 |
A zero-copy multiprocess dataloader for JAX. |
2024-03-24 19:43:42 |
xminigrid |
0.7.0 |
JAX-accelerated meta-reinforcement learning environments inspired by XLand and MiniGrid |
2024-03-24 11:00:37 |
gpjax |
0.8.2 |
Gaussian processes in JAX. |
2024-03-23 09:28:04 |
inox |
0.5.0 |
Stainless neural networks in JAX |
2024-03-21 12:20:24 |
diffusers |
0.27.2 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-03-20 01:54:25 |
etils |
1.8.0 |
Collection of common python utils |
2024-03-19 15:36:31 |
dataclass-array |
1.5.2 |
Dataclasses that behave like numpy arrays (with indexing, slicing, vectorization). |
2024-03-19 15:25:17 |
chex |
0.1.86 |
Chex: Testing made fun, in JAX! |
2024-03-19 10:05:29 |
brax |
0.10.3 |
A differentiable physics engine written in JAX. |
2024-03-13 20:26:39 |
feedbax |
0.1.2 |
Optimal feedback control + interventions in JAX. |
2024-03-13 14:44:59 |
differt |
0.0.10 |
Differentiable Ray Tracing Toolbox for Radio Propagation Simulations |
2024-03-12 18:32:43 |
tsnex |
0.0.1 |
Minimal t-distributed stochastic neighbor embedding (t-SNE) implementation in JAX. |
2024-03-12 17:34:37 |
nanodl |
1.2.1.dev1 |
A Jax-based library for designing and training transformer models from scratch. |
2024-03-12 09:51:11 |
jax-chmc |
0.0.7 |
An implementation of a family of MCMC methods on implicitly defined manifolds |
2024-03-09 09:10:42 |
autoray |
0.6.9 |
Abstract your array operations. |
2024-03-09 07:47:57 |
jaxtyping |
0.2.28 |
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. |
2024-03-07 17:31:29 |
cfrx |
0.0.2 |
Counterfactual Regret Minimization in Jax |
2024-03-05 14:45:34 |