Roy Frostig
a60ead6fd1
enable partitionable threefry by default
...
PiperOrigin-RevId: 715242560
2025-01-13 22:46:24 -08:00
Dan Foreman-Mackey
96268dcae6
Fix dtype bug in jax.scipy.fft.idct
2024-09-25 12:55:43 -04:00
Jake VanderPlas
2efd1ec011
jax.scipy.fft.dct: implement & test norm='backward'
2024-07-22 11:18:35 -07:00
Jake VanderPlas
326559ca47
jax.scipy.fft: error for unsupported norm argument
2024-07-22 10:32:03 -07:00
Dan Foreman-Mackey
6d35b109fd
Rename "Example" to "Examples" in docstrings.
...
This PR updates all docstrings that previously had a section heading
called "Example" and replaces that with "Examples" to be consistent.
2024-06-21 11:43:16 -04:00
jax authors
bf59a67bf0
Merge pull request #21657 from rajasekharporeddy:test_branch7
...
PiperOrigin-RevId: 640652642
2024-06-05 14:35:35 -07:00
rajasekharporeddy
be41f309ed
Add code examples to jax.scipy.fft.idct and jax.scipy.fft.idctn docs
2024-06-06 01:45:07 +05:30
jax authors
913ff50000
Merge pull request #21418 from rajasekharporeddy:test_branch6
...
PiperOrigin-RevId: 640617948
2024-06-05 12:53:17 -07:00
rajasekharporeddy
bb6cb2c0de
Add code examples to jax.scipy.fft.dct doc
2024-06-04 22:10:27 +05:30
rajasekharporeddy
0df26689bd
Add code example to jax.scipy.fft.dctn doc
2024-06-04 21:41:46 +05:30
Jake VanderPlas
75921162ab
jax.scipy.fft: manually document functions to avoid scipy import
2024-04-29 09:23:43 -07:00
Jake VanderPlas
f35538be6a
DOC: fix two minor doc issues
2024-03-05 15:22:42 -08:00
Jake VanderPlas
43a9faa06a
Rename _wraps to implements
2024-01-24 14:14:19 -08:00
Sergei Lebedev
f936613b06
Upgrade remaining sources to Python 3.9
...
This PR is a follow up to #18881 .
The changes were generated by adding
from __future__ import annotations
to the files which did not already have them and running
pyupgrade --py39-plus --keep-percent-format {jax,tests,jaxlib,examples,benchmarks}/**/*.py
2023-12-13 10:29:45 +00:00
Neil Girdhar
3c920c0120
Switch from flake8 to Ruff
2023-11-15 22:35:52 -05:00
Sergei Lebedev
df7f6a06c0
MAINT Use a generator expression in tuple([... for ... in ...])
...
In a few cases I also replaced tuple([*xs, *ys]) with (*xs, ys), because
tuple literals support unpacking as well.
2023-09-21 22:25:38 +01:00
Peter Hawkins
319ab98980
Apply pyupgrade --py39-plus.
...
Notable changes:
* use PEP 585 type names
* use PEP 604 type union syntax where `from __future__ import annotations` is present.
* use f-strings in more places.
* remove redundant arguments to open().
2023-07-21 14:49:44 -04:00
Vaishaal Shankar
add15aca25
implement idct and idctn + add function to scipy.rst
2023-04-17 12:12:51 -07:00
Jake VanderPlas
760deb310e
Remove leading underscores in jax._src.numpy.util
2023-03-13 12:18:36 -07:00
Jake VanderPlas
76e4a1d40f
[typing] add annotations to jax.scipy.fft
2022-10-09 05:18:45 -07:00
Jake VanderPlas
730f9930aa
Update scipy version in jax.scipy.fft
2022-10-07 09:57:39 -07:00
Peter Hawkins
ba557d5e1b
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
...
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
2022-09-22 12:27:19 -07:00
Jake VanderPlas
297a2969a5
[x64] make fft functionality compatible with strict dtype promotion
2022-06-15 10:10:44 -07:00
Jake VanderPlas
f6e3f1b4ad
Cleanup: remove duplicate canonicalize_axis utility
2021-11-23 16:54:02 -08:00
Peter Hawkins
8b2123968a
Switch internal users of jax.util.partial to use functools.partial.
2021-09-13 21:09:58 -04:00
Julius Kunze
6d83027b69
Support scipy.fft.dct/dctn type=2
2021-08-17 18:56:44 +02:00
